From fe1fcef949bf2e84a6e8dc67fb4e5e62dded6336 Mon Sep 17 00:00:00 2001 From: Silver <65376327+silveroxides@users.noreply.github.com> Date: Sun, 11 May 2025 21:44:16 +0200 Subject: [PATCH] Update __init__.py --- __init__.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/__init__.py b/__init__.py index b258d9c..b07f35a 100644 --- a/__init__.py +++ b/__init__.py @@ -4,7 +4,7 @@ import logging import numpy from comfy.samplers import SchedulerHandler, SCHEDULER_HANDLERS, SCHEDULER_NAMES -def sigmoid_offset_scheduler(model_sampling, steps: int, square_k: float = 1.2, base_c: float = 0.6) -> torch.Tensor: +def sigmoid_offset_scheduler(model_sampling, steps: int, square_k: float = 1.0, base_c: float = 0.5) -> torch.Tensor: if not hasattr(model_sampling, 'sigmas') or len(model_sampling.sigmas) < 2: logging.error("model_sampling object must have a 'sigmas' attribute with at least two values.") @@ -56,7 +56,7 @@ def sigmoid_offset_scheduler(model_sampling, steps: int, square_k: float = 1.2, last_t = t # Add final 0.0 - # sigs.append(0.0) + sigs.append(0.0) return torch.FloatTensor(sigs) @@ -73,8 +73,8 @@ class SigmoidOffsetScheduler: return {"required": {"model": ("MODEL",), "steps": ("INT", {"default": 30, "min": 1, "max": 10000}), - "square_k": ("FLOAT", {"default": 1.2, "min": 0.0, "max": 5.0, "step":0.01, "round": False, "tooltip": "Sigmoid steepness factor. Higher values = steeper transition."}), - "base_c": ("FLOAT", {"default": 0.6, "min": -5.0, "max": 5.0, "step":0.01, "round": False, "tooltip": "Control parameter (0.0-1.0) that shifts sigmoid curve. '< 0.5: More time at low sigmas (early denoising)' and '> 0.5: More time at high sigmas (late denoising)'."}), + "square_k": ("FLOAT", {"default": 1.0, "min": 0.0, "max": 5.0, "step":0.01, "round": False, "tooltip": "Sigmoid steepness factor. Higher values = steeper transition."}), + "base_c": ("FLOAT", {"default": 0.5, "min": -5.0, "max": 5.0, "step":0.01, "round": False, "tooltip": "Control parameter (0.0-1.0) that shifts sigmoid curve. '< 0.5: More time at low sigmas (early denoising)' and '> 0.5: More time at high sigmas (late denoising)'."}), } } RETURN_TYPES = ("SIGMAS",) @@ -90,4 +90,4 @@ class SigmoidOffsetScheduler: NODE_CLASS_MAPPINGS = { "SigmoidOffsetScheduler": SigmoidOffsetScheduler, -} \ No newline at end of file +}