transformers/tests/models/llava_next
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 Add LLaVa-1.6, bis (#29586) 2024-03-20 15:51:12 +00:00
test_image_processor_llava_next.py Add LLaVa-1.6, bis (#29586) 2024-03-20 15:51:12 +00:00
test_modeling_llava_next.py Fix llava half precision and autocast issues (#29721) 2024-05-01 17:49:44 +01:00