Compare commits

..
4 Commits
Author SHA1 Message Date
皓童 af5cf1ccc0 modify edit 2024-12-08 00:31:50 +08:00
皓童 0ed54e05e4 modify edit 2024-12-08 00:28:11 +08:00
皓童 27a2b7833f modify edit 2024-12-08 00:22:43 +08:00
皓童 6c8ce55888 modify edit 2024-12-08 00:16:03 +08:00
6 changed files with 6 additions and 8 deletions
+2 -2
View File
@@ -2,7 +2,7 @@ albumentations
beautifulsoup4
bezier
einops
modelscope[framework]
modelscope[framework]<=1.20.1
ms-swift
numpy
open_clip_torch
@@ -15,4 +15,4 @@ scikit-image
scikit-learn
sentencepiece
torchsde
transformers
transformers<=4.46.3
+1 -1
View File
@@ -1,4 +1,4 @@
bitsandbytes
bitsandbytes<=0.44.1
gradio
gradio_imageslider
imagehash
View File
+1 -3
View File
@@ -27,7 +27,6 @@ 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
@@ -67,8 +66,7 @@ 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(
os.path.dirname(scepter.dirname), self.model_cfg_dir,
self.model_yamls = glob.glob(os.path.join(self.model_cfg_dir,
'*.yaml'))
self.model_choices = dict()
self.default_model_name = ''
+1 -1
View File
@@ -166,7 +166,7 @@ class ModelNode:
input_data.update({'mask': mask})
if parameters:
seed = parameters.get('random_seed', -1)
seed = parameters.pop('seed', -1)
input_params.update({'seed': seed})
input_data.update(parameters)
+1 -1
View File
@@ -59,6 +59,6 @@ class ParameterNode:
'guide_rescale': guide_rescale,
'discretization': discretization,
'target_size_as_tuple': [output_height, output_width],
'random_seed': random_seed
'seed': random_seed
}
return (out, )