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