Skip to content

Commit 090b391

Browse files
committed
Convert sigmas to np.array in FlowMatch set_timesteps
1 parent c44fba8 commit 090b391

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

src/diffusers/schedulers/scheduling_flow_match_euler_discrete.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -207,6 +207,7 @@ def set_timesteps(
207207

208208
sigmas = timesteps / self.config.num_train_timesteps
209209
else:
210+
sigmas = np.array(sigmas).astype(np.float32)
210211
num_inference_steps = len(sigmas)
211212
self.num_inference_steps = num_inference_steps
212213

0 commit comments

Comments
 (0)