Merge pull request THUDM#354 from ildar-idrisov/main

typo with the path to hf
ngoiyaeric · Jan 29, 2024 · 7b8240f · 7b8240f
2 parents 1d9295a + 0b1ce7f
commit 7b8240f
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/composite_demo/client.py b/composite_demo/client.py
@@ -26,15 +26,15 @@
         'path': os.environ.get('TOKENIZER_PATH', 'lmsys/vicuna-7b-v1.5'),
     },
     'agent_chat': {
-        'path': os.environ.get('MODEL_PATH_AGENT_CHAT', 'THDUM/cogagent-chat-hf'),
+        'path': os.environ.get('MODEL_PATH_AGENT_CHAT', 'THUDM/cogagent-chat-hf'),
         'device': ['cuda:0']
     },
     'vlm_chat': {
-        'path': os.environ.get('MODEL_PATH_VLM_CHAT', 'THDUM/cogvlm-chat-hf'),
+        'path': os.environ.get('MODEL_PATH_VLM_CHAT', 'THUDM/cogvlm-chat-hf'),
         'device': ['cuda:3']
     },
     'vlm_grounding': {
-        'path': os.environ.get('MODEL_PATH_VLM_GROUNDING','THDUM/cogvlm-grounding-generalist-hf'),
+        'path': os.environ.get('MODEL_PATH_VLM_GROUNDING','THUDM/cogvlm-grounding-generalist-hf'),
         'device': ['cuda:6']
     }
 }