From 0b1ce7fa7496593c91a2383b4ee1575c0e9dec57 Mon Sep 17 00:00:00 2001 From: ildar-idrisov <34361994+ildar-idrisov@users.noreply.github.com> Date: Mon, 29 Jan 2024 15:44:44 +0700 Subject: [PATCH] typo with the path to hf --- composite_demo/client.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/composite_demo/client.py b/composite_demo/client.py index d5a30b9c..cdbe6f0d 100644 --- 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'] } }