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

30f2e9bd · hlky · GitHub · 2312b27f · 30f2e9bd
Unverified Commit 30f2e9bd authored Dec 03, 2024 by hlky Committed by GitHub Dec 02, 2024
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 0 deletions

src/diffusers/schedulers/scheduling_flow_match_euler_discrete.py ...fusers/schedulers/scheduling_flow_match_euler_discrete.py +1 -0

No files found.
--- a/src/diffusers/schedulers/scheduling_flow_match_euler_discrete.py
+++ b/src/diffusers/schedulers/scheduling_flow_match_euler_discrete.py
@@ -207,6 +207,7 @@ class FlowMatchEulerDiscreteScheduler(SchedulerMixin, ConfigMixin):

            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