diff --git a/nodes.py b/nodes.py index 86e4fed..c431f19 100644 --- a/nodes.py +++ b/nodes.py @@ -194,10 +194,10 @@ class DiscoDiffusion: "n_batches": ("INT", {"default": 1, "min": 1, "max": 16}), # "max_frames": ("INT", {"default": 1, "min": 1, "max": 1000}), "sampling_mode": (["plms", "ddim", "stsp", "ltsp"], {"default": "ddim"}), - "clip_guidance_scale": ("INT", { "default": 5000, "min": 1, "max": 10000000 }), - "tv_scale": ("INT", { "default": 0, "min": 0, "max": 100000 }), - "range_scale": ("INT", { "default": 150, "min": 0, "max": 100000 }), - "sat_scale": ("INT", { "default": 0, "min": 0, "max": 100000 }), + "clip_guidance_scale": ("FLOAT", { "default": 5000, "min": 1, "max": 10000000 }), + "tv_scale": ("FLOAT", { "default": 0, "min": 0, "max": 100000 }), + "range_scale": ("FLOAT", { "default": 150, "min": 0, "max": 100000 }), + "sat_scale": ("FLOAT", { "default": 0, "min": 0, "max": 100000 }), }, "optional": { "extra_settings": ("DISCO_DIFFUSION_EXTRA_SETTINGS",),