Allow for weight sharing between Qmap and Vmap #5
Description
Refactor particularly the CrossAttention code to make it use a variant QKVmap with two inputs, which would allow for implementation of QKVmaps with weight sharing
{{ message }}
Refactor particularly the CrossAttention code to make it use a variant QKVmap with two inputs, which would allow for implementation of QKVmaps with weight sharing