transformers/tests/models/llava
laurentd-lunit 0f49deacbf
[feat] LlavaNext add feature size check to avoid CUDA Runtime Error (#33608)
* [feat] add feature size check to avoid CUDA Runtime Error

* [minor] add error handling to all llava models

* [minor] avoid nested if else

* [minor] add error message to Qwen2-vl and chameleon

* [fix] token dimension for check

* [minor] add feature dim check for videos too

* [fix] dimension check

* [fix] test reference values

---------

Co-authored-by: Raushan Turganbay <raushan@huggingface.co>
2024-10-15 16:19:18 +02:00
..
__init__.py [Llava] Add Llava to transformers (#27662) 2023-12-07 09:30:47 +01:00
test_modeling_llava.py [feat] LlavaNext add feature size check to avoid CUDA Runtime Error (#33608) 2024-10-15 16:19:18 +02:00
test_processor_llava.py Add support for args to ProcessorMixin for backward compatibility (#33479) 2024-09-20 11:40:59 -04:00