transformers/tests/models/cohere
Michael Benayoun 17cd7a9d28
Fix `torch.fx` symbolic tracing for LLama (#30047)
* [WIP] fix fx

* [WIP] fix fx

* [WIP] fix fx

* [WIP] fix fx

* [WIP] fix fx

* Apply changes to other models
2024-04-05 15:14:09 +02:00
..
__init__.py Cohere Model Release (#29622) 2024-03-15 14:29:11 +01:00
test_modeling_cohere.py Fix `torch.fx` symbolic tracing for LLama (#30047) 2024-04-05 15:14:09 +02:00
test_tokenization_cohere.py Cohere Model Release (#29622) 2024-03-15 14:29:11 +01:00