zhangyiss/mlx-examples
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx-examples.git synced 2025-09-01 21:01:32 +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
1 changed files with 1 additions and 1 deletions
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()