fxmarty a6adc05e6b symbolic_trace: add past_key_values, llama, sdpa support (#28447) ...

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

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

* remove spaces

* trigger ci

* use explicit variable names

2024-01-17 11:50:53 +01:00

..

__init__.py

LLaMA Implementation (#21955)

2023-03-16 09:00:53 -04:00

test_modeling_flax_llama.py

Add Llama Flax Implementation (#24587)

2023-12-07 07:05:00 +01:00

test_modeling_llama.py

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

2024-01-17 11:50:53 +01:00

test_tokenization_llama.py

[Styling] stylify using ruff (#27144)

2023-11-16 17:43:19 +01:00