From d7dbdc529280cb297d82cc1d7efcdbe9d9bcf305 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E7=9A=93=E7=AB=A5?= Date: Thu, 5 Dec 2024 16:05:30 +0800 Subject: [PATCH] modify default path --- scepter/studio/chatbot/chatbot.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/scepter/studio/chatbot/chatbot.py b/scepter/studio/chatbot/chatbot.py index c94c88f..5510e82 100644 --- a/scepter/studio/chatbot/chatbot.py +++ b/scepter/studio/chatbot/chatbot.py @@ -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 = ''