Make timestep ranges more usable on rectified flow models.

This breaks some old workflows but should make the nodes actually useful.
comfyanonymous · Dec 5, 2024 · 1e21f4c · 1e21f4c
1 parent 9a616b8
commit 1e21f4c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/comfy/model_sampling.py b/comfy/model_sampling.py
@@ -243,7 +243,7 @@ def percent_to_sigma(self, percent):
             return 1.0
         if percent >= 1.0:
             return 0.0
-        return 1.0 - percent
+        return time_snr_shift(self.shift, 1.0 - percent)
 
 class StableCascadeSampling(ModelSamplingDiscrete):
     def __init__(self, model_config=None):
@@ -336,4 +336,4 @@ def percent_to_sigma(self, percent):
             return 1.0
         if percent >= 1.0:
             return 0.0
-        return 1.0 - percent
+        return flux_time_shift(self.shift, 1.0, 1.0 - percent)