zhangyiss/mlx-examples
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx-examples.git synced 2025-06-24 17:31:18 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fix bug in FluxSampler.timesteps method (#1131)

Browse Source
This commit is contained in:
hehua2008 2024-12-03 05:15:19 +08:00 committed by GitHub
parent 8801beb66f
commit 2a9294a5f0
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
flux/flux/sampler.py
View File

@ -25,7 +25,7 @@ class FluxSampler:
): ):
t = mx.linspace(start, stop, num_steps + 1) t = mx.linspace(start, stop, num_steps + 1)
if self._schnell: if not self._schnell:
t = self._time_shift(image_sequence_length, t) t = self._time_shift(image_sequence_length, t)
return t.tolist() return t.tolist()