test

AllentDan · May 9, 2024 · b50d2ac · b50d2ac
1 parent 6688a5c
commit b50d2ac
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/lmdeploy/vl/model/builder.py b/lmdeploy/vl/model/builder.py
@@ -70,4 +70,8 @@ def vl_model_with_tokenizer(model_path: str, device: str):
             return LlavaVisionModel.model_with_tokenizer(model_path, device)
     if arch in ['MiniGeminiLlamaForCausalLM', 'MGMLlamaForCausalLM']:
         return MiniGeminiVisionModel.model_with_tokenizer(model_path, device)
+    if arch == 'InternLMXComposer2ForCausalLM':
+        return Xcomposer2VisionModel.model_with_tokenizer(model_path)
+    if arch == 'InternVLChatModel':
+        return InternVLVisionModel(model_path)
     raise ValueError(f'unsupported vl model with arch {arch}')