transformers

Shape cross-attention keys and values by the states they come from

#48356

Open