Convert sigmas to np.array in FlowMatch set_timesteps (#10088)

huggingface · Dec 3, 2024 · 30f2e9b · 30f2e9b
1 parent 2312b27
commit 30f2e9b
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/diffusers/schedulers/scheduling_flow_match_euler_discrete.py b/src/diffusers/schedulers/scheduling_flow_match_euler_discrete.py
@@ -207,6 +207,7 @@ def set_timesteps(
 
             sigmas = timesteps / self.config.num_train_timesteps
         else:
+            sigmas = np.array(sigmas).astype(np.float32)
             num_inference_steps = len(sigmas)
         self.num_inference_steps = num_inference_steps