From 7ee0a5570531f56f77c1956ca8135a04cb566470 Mon Sep 17 00:00:00 2001
From: hehua2008 <hegan2010@163.com>
Date: Tue, 3 Dec 2024 05:15:19 +0800
Subject: [PATCH] Fix bug in FluxSampler.timesteps method (#1131)

---
 flux/flux/sampler.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/flux/flux/sampler.py b/flux/flux/sampler.py
index 3bff1ca2..54c4fe35 100644
--- a/flux/flux/sampler.py
+++ b/flux/flux/sampler.py
@@ -25,7 +25,7 @@ class FluxSampler:
     ):
         t = mx.linspace(start, stop, num_steps + 1)
 
-        if self._schnell:
+        if not self._schnell:
             t = self._time_shift(image_sequence_length, t)
 
         return t.tolist()