transformers
[feat] LlavaNext add feature size check to avoid CUDA Runtime Error
#33608
Merged
Merged
[feat] LlavaNext add feature size check to avoid CUDA Runtime Error #33608
zucchini-nlp
merged
11 commits into
huggingface:main
from
laurentd-lunit:feat/llava-next-feature-token-mismatch-error-handling
zucchini-nlp
commented
on 2024-09-23
zucchini-nlp
approved these changes
on 2024-09-27
[feat] add feature size check to avoid CUDA Runtime Error
4ea3e0ed
[minor] add error handling to all llava models
610bc752
[minor] avoid nested if else
353c610e
[minor] add error message to Qwen2-vl and chameleon
53d407df
[fix] token dimension for check
fc076402
[minor] add feature dim check for videos too
9ac66896
[fix] dimension check
edab0fc8
[fix] test reference values
bea8910b
LysandreJik
approved these changes
on 2024-09-30
ArthurZucker
approved these changes
on 2024-10-05
Merge branch 'main' into feat/llava-next-feature-token-mismatch-error…
028965d4
Merge remote-tracking branch 'upstream/main' into feat/llava-next-fea…
244693f6
Merge remote-tracking branch 'upstream/main' into feat/llava-next-fea…
82058f2c
Reviewers
Assignees
No one assigned
Labels
None yet
Milestone
No milestone
Login to write a write a comment.
Login via GitHub