Fixes
This commit is contained in:
+2
-2
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user