From d459cabea48cb8ecdbf226b9626511d86550d648 Mon Sep 17 00:00:00 2001
From: hehua2008 <hegan2010@gmail.com>
Date: Sun, 1 Dec 2024 13:07:48 +0800
Subject: [PATCH] Fix bug in FluxSampler.timesteps method

---
 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()