tests/models/llama/test_modeling_llama.py · a6adc05e6b10479474ecdb1dbd3f9c6925d5e332 · chenpangpang / transformers

symbolic_trace: add past_key_values, llama, sdpa support (#28447) · a6adc05e

fxmarty authored Jan 17, 2024

* torch.fx: add pkv, llama, sdpa support

* Update src/transformers/models/opt/modeling_opt.py

* remove spaces

* trigger ci

* use explicit variable names

a6adc05e

test_modeling_llama.py 29.9 KB

Replace test_modeling_llama.py