transformers/tests/models/mistral
fxmarty 64845307b3
Remove unnecessary unsqueeze - squeeze in rotary positional embedding (#26162)
* remove unnecessary unsqueeze-squeeze in llama

* correct other models

* fix

* revert gpt_neox_japanese

* fix copie

* fix test
2023-10-06 18:25:15 +09:00
..
__init__.py [Mistral] Mistral-7B-v0.1 support (#26447) 2023-09-27 18:30:46 +02:00
test_modeling_mistral.py Remove unnecessary unsqueeze - squeeze in rotary positional embedding (#26162) 2023-10-06 18:25:15 +09:00