zhangyiss/mlx-examples
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx-examples.git synced 2025-08-29 03:01:34 +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 Billel Mokeddem
parent 278884f6ca
commit 7ee0a55705
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)
if self._schnell:
if not self._schnell:
t = self._time_shift(image_sequence_length, t)
return t.tolist()