From 30f2e9bd202c89bb3862c8ada470d0d1ac8ee0e5 Mon Sep 17 00:00:00 2001 From: hlky Date: Tue, 3 Dec 2024 00:18:40 +0000 Subject: [PATCH] Convert `sigmas` to `np.array` in FlowMatch set_timesteps (#10088) --- src/diffusers/schedulers/scheduling_flow_match_euler_discrete.py | 1 + 1 file changed, 1 insertion(+) diff --git a/src/diffusers/schedulers/scheduling_flow_match_euler_discrete.py b/src/diffusers/schedulers/scheduling_flow_match_euler_discrete.py index d01071ec27b8..91264e805a0f 100644 --- 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