@@ -106,12 +106,12 @@ def check_vl_llm(config: dict) -> bool:
106
106
107
107
arch = config ['architectures' ][0 ]
108
108
supported_archs = set ([
109
- 'LlavaLlamaForCausalLM' , 'LlavaMistralForCausalLM' , 'CogVLMForCausalLM ' , 'InternLMXComposer2ForCausalLM ' ,
110
- 'InternVLChatModel ' , 'MiniCPMV ' , 'LlavaForConditionalGeneration ' , 'LlavaNextForConditionalGeneration ' ,
111
- 'Phi3VForCausalLM ' , 'Qwen2VLForConditionalGeneration ' , 'Qwen2_5_VLForConditionalGeneration ' ,
112
- 'MllamaForConditionalGeneration ' , 'MolmoForCausalLM ' , 'Gemma3ForConditionalGeneration ' ,
113
- 'Llama4ForConditionalGeneration ' , 'InternVLForConditionalGeneration ' , 'InternS1ForConditionalGeneration ' ,
114
- 'Glm4vForConditionalGeneration ' , 'LlavaQwenForCausalLM '
109
+ 'LlavaLlamaForCausalLM' , 'LlavaMistralForCausalLM' , 'LlavaQwenForCausalLM ' , 'CogVLMForCausalLM ' ,
110
+ 'InternLMXComposer2ForCausalLM ' , 'InternVLChatModel ' , 'MiniCPMV ' , 'LlavaForConditionalGeneration ' ,
111
+ 'LlavaNextForConditionalGeneration ' , 'Phi3VForCausalLM ' , 'Qwen2VLForConditionalGeneration ' ,
112
+ 'Qwen2_5_VLForConditionalGeneration ' , 'MllamaForConditionalGeneration ' , 'MolmoForCausalLM ' ,
113
+ 'Gemma3ForConditionalGeneration ' , 'Llama4ForConditionalGeneration ' , 'InternVLForConditionalGeneration ' ,
114
+ 'InternS1ForConditionalGeneration ' , 'Glm4vForConditionalGeneration '
115
115
])
116
116
if arch == 'QWenLMHeadModel' and 'visual' in config :
117
117
return True
0 commit comments