webdemo openxlab

eudjdh · Jul 15, 2024 · 7592104 · 7592104
1 parent e8b2e69
commit 7592104
Showing 1 changed file with 8 additions and 3 deletions.
diff --git a/web_internlm2_5.py b/web_internlm2_5.py
@@ -15,7 +15,7 @@
 Using `python path/to/web_demo.py` may cause unknown problems.
 """
 # isort: skip_file
-import copy
+import copy, os
 import warnings
 from dataclasses import asdict, dataclass
 from typing import Callable, List, Optional
@@ -31,10 +31,15 @@
 
 logger = logging.get_logger(__name__)
 
-# local
-model_path = '/root/EmoLLM/xtuner_config/hf4'
+# # local
+# model_path = '/root/EmoLLM/xtuner_config/hf_safe'
+
 # Online downloading will be added later
 
+model_path = './EmoLLM_V3.0'
+os.system(f'git clone https://code.openxlab.org.cn/chg0901/EmoLLM_V3.0.git {model_path}')
+os.system(f'cd {model_path} && git lfs pull')
+
 @dataclass
 class GenerationConfig:
     # this config is used for chat to provide more diversity