From 090b3910e8f36a1ac07db722f2f14effb948b0be Mon Sep 17 00:00:00 2001 From: hlky Date: Mon, 2 Dec 2024 21:59:54 +0000 Subject: [PATCH] Convert `sigmas` to `np.array` in FlowMatch set_timesteps --- 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