Skip to content

Commit 621bb3b

Browse files
committed
update code
1 parent 5f24a68 commit 621bb3b

File tree

1 file changed

+6
-6
lines changed

1 file changed

+6
-6
lines changed

lmdeploy/archs.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -106,12 +106,12 @@ def check_vl_llm(config: dict) -> bool:
106106

107107
arch = config['architectures'][0]
108108
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'
115115
])
116116
if arch == 'QWenLMHeadModel' and 'visual' in config:
117117
return True

0 commit comments

Comments
 (0)