zhangyiss/mlx-examples
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx-examples.git synced 2025-08-30 10:56:38 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fix bug in FluxSampler.timesteps method

Browse Source
This commit is contained in:
hehua2008 2024-12-01 13:07:48 +08:00
parent cefe793ae0
commit d459cabea4
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()