SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

[VLM] fix loading issues (#38051)

Browse Source 
* fix qwen2-vl loading

* fix a few nore models

* delete print

* fix copies
This commit is contained in:
Raushan Turganbay
2025-05-12 12:14:04 +02:00
committed by GitHub
parent a31fa218ad
commit a5c6172c81
5 changed files with 9 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
src/transformers/modeling_utils.py
View File

@@ -219,22 +219,19 @@ TORCH_INIT_FUNCTIONS = {
# DO NOT MODIFY, KEPT FOR BC ONLY
VLMS = [
"aria",
"aya_vision",
"ayavision",
"emu3",
"fuyu",
"got_ocr2",
"gotocr2",
"gemma3",
"internvl",
"llava",
"llava_next",
"llava_next_video",
"llava_onevision",
"llava", # all llava prefixed models fall under this check
"mistral3",
"mllama",
"paligemma",
"qwen2_vl",
"qwem2_5_vl",
"video_llava",
"qwen2vl",
"qwen2_5_vl",
"videollava",
"vipllava",
]