modify default path

Merge pull request #69 from modelscope/v1.3.0_dev
add init file for chatbot
2024-12-05 16:05:30 +08:00 · 2024-12-05 15:23:37 +08:00 · 2024-11-26 14:58:52 +08:00 · 2024-11-26 14:58:36 +08:00 · 2024-11-26 14:57:20 +08:00 · 2024-11-26 14:55:29 +08:00
3 changed files with 5 additions and 3 deletions
@@ -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 = ''
@@ -166,7 +166,7 @@ class ModelNode:
            input_data.update({'mask': mask})

        if parameters:
-            seed = parameters.pop('seed', -1)
+            seed = parameters.get('random_seed', -1)
            input_params.update({'seed': seed})
            input_data.update(parameters)

@@ -59,6 +59,6 @@ class ParameterNode:
            'guide_rescale': guide_rescale,
            'discretization': discretization,
            'target_size_as_tuple': [output_height, output_width],
-            'seed': random_seed
+            'random_seed': random_seed
        }
        return (out, )
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
jiangzeyinzi	adda36e39d	Merge pull request #66 from yaosheng216/patch-7 Update model_node.py	2024-11-26 14:58:52 +08:00
jiangzeyinzi	1da1864993	Merge pull request #65 from yaosheng216/patch-6 Update parameter_node.py	2024-11-26 14:58:36 +08:00
Great	5eac362325	Update model_node.py	2024-11-26 14:57:20 +08:00
Great	9ae3bca43d	Update parameter_node.py	2024-11-26 14:55:29 +08:00
jiangzeyinzi	ca034ef765	Merge pull request #64 from modelscope/v1.3.0_dev V1.3.0 dev	2024-11-26 12:13:08 +08:00