From 1e21f4c14e6a0ce68764e71170082ed2c2da5c79 Mon Sep 17 00:00:00 2001 From: comfyanonymous Date: Thu, 5 Dec 2024 16:40:58 -0500 Subject: [PATCH] Make timestep ranges more usable on rectified flow models. This breaks some old workflows but should make the nodes actually useful. --- comfy/model_sampling.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/comfy/model_sampling.py b/comfy/model_sampling.py index 8b4e095d..4370516b 100644 --- a/comfy/model_sampling.py +++ b/comfy/model_sampling.py @@ -243,7 +243,7 @@ class ModelSamplingDiscreteFlow(torch.nn.Module): 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 @@ class ModelSamplingFlux(torch.nn.Module): 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)