Merge pull request #66 from yaosheng216/patch-7

Update model_node.py

requirements/framework.txt

@@ -2,7 +2,7 @@ albumentations
 beautifulsoup4
 bezier
 einops
-modelscope[framework]<=1.20.1
+modelscope[framework]
 ms-swift
 numpy
 open_clip_torch
@@ -15,4 +15,4 @@ scikit-image
 scikit-learn
 sentencepiece
 torchsde
-transformers<=4.46.3
+transformers

requirements/scepter_studio.txt

@@ -1,4 +1,4 @@
-bitsandbytes<=0.44.1
+bitsandbytes
 gradio
 gradio_imageslider
 imagehash

scepter/workflow/model_node.py

@@ -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)
 

scepter/workflow/parameter_node.py

@@ -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, )