This commit is contained in:
asagi4
2023-12-06 14:07:48 +02:00
parent 788ffb6572
commit 59d67f304b
+2 -2
View File
@@ -21,9 +21,9 @@ class CADS:
"noise_scale": ("FLOAT", {"min": 0.0, "max": 1.0, "step": 0.01, "default": 0.25}),
"t1": ("FLOAT", {"min": 0.0, "max": 1.0, "step": 0.01, "default": 0.6}),
"t2": ("FLOAT", {"min": 0.0, "max": 1.0, "step": 0.01, "default": 0.9}),
"rescale": ("FLOAT", {"min": 0.0, "max": 1.0, "step": 0.01, "default": 0.0}),
},
"optional": {
"rescale": ("FLOAT", {"min": 0.0, "max": 1.0, "step": 0.01, "default": 0.0}),
"start_step": ("INT", {"min": -1, "max": 10000, "default": -1}),
"total_steps": ("INT", {"min": -1, "max": 10000, "default": -1}),
},
@@ -34,7 +34,7 @@ class CADS:
CATEGORY = "utils"
def do(self, model, noise_scale, t1, t2, rescale, start_step=0, total_steps=0):
def do(self, model, noise_scale, t1, t2, rescale=0.0, start_step=-1, total_steps=-1):
previous_wrapper = model.model_options.get("model_function_wrapper")
im = model.model.model_sampling