transformers
0f49deac - [feat] LlavaNext add feature size check to avoid CUDA Runtime Error (#33608)

Commit
1 year ago
[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>
Parents
Loading