Compare commits

..
3 Commits
Author SHA1 Message Date
皓童 d7dbdc5292 modify default path 2024-12-05 16:05:30 +08:00
mcj 711c10a68c Merge pull request #69 from modelscope/v1.3.0_dev
add init file for chatbot
2024-12-05 15:23:37 +08:00
皓童 448cdba522 add init file for chatbot 2024-12-05 15:20:00 +08:00
2 changed files with 3 additions and 1 deletions
View File
+3 -1
View File
@@ -27,6 +27,7 @@ from scepter.modules.utils.config import Config
from scepter.modules.utils.directory import get_md5
from scepter.modules.utils.file_system import FS
from scepter.studio.utils.env import init_env
import scepter
from importlib.metadata import version
from .example import get_examples
@@ -66,7 +67,8 @@ class ChatBotUI(object):
self.cache_dir = cfg.WORK_DIR
self.chatbot_examples = get_examples(self.cache_dir) if not cfg.get('SKIP_EXAMPLES', False) else []
self.model_cfg_dir = cfg.MODEL.EDIT_MODEL.MODEL_CFG_DIR
self.model_yamls = glob.glob(os.path.join(self.model_cfg_dir,
self.model_yamls = glob.glob(os.path.join(
os.path.dirname(scepter.dirname), self.model_cfg_dir,
'*.yaml'))
self.model_choices = dict()
self.default_model_name = ''