add init file for chatbot

2024-12-05 15:20:00 +08:00
3 changed files with 2 additions and 2 deletions
@@ -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)

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