transformers/tests/models/llava
Fraser Mince 5090ea3f68
Fix llava half precision and autocast issues (#29721)
* Ensure input_embeds and image_features are the same dtype in autocast

* Fix nans in half precision llava-next and fix autocasting behavior.

* Fix styling issues.

* fix randn newline instantiation

* fix broken slow llava test

* Fix llava next init.

* fix styling issues

* [run-slow]llava,llava_next

* fix styling issues
2024-05-01 17:49:44 +01:00
..
__init__.py [`Llava`] Add Llava to transformers (#27662) 2023-12-07 09:30:47 +01:00
test_modeling_llava.py Fix llava half precision and autocast issues (#29721) 2024-05-01 17:49:44 +01:00