transformers/tests/models/mistral
Yanan Xie b09912c8f4
Fix mistral generate for long prompt / response (#27548)
* Fix mistral generate for long prompt / response

* Add unit test

* fix linter

* fix linter

* fix test

* add assisted generation test for mistral and load the model in 4 bit + fa2
2023-11-27 10:18:41 +01:00
..
__init__.py [Mistral] Mistral-7B-v0.1 support (#26447) 2023-09-27 18:30:46 +02:00
test_modeling_mistral.py Fix mistral generate for long prompt / response (#27548) 2023-11-27 10:18:41 +01:00