From 98a52db0583dc2e964926a7e3ce646d6b74e66ec Mon Sep 17 00:00:00 2001 From: hlky Date: Sat, 23 Nov 2024 15:55:56 +0000 Subject: [PATCH] timesteps --- src/diffusers/schedulers/scheduling_euler_discrete.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/diffusers/schedulers/scheduling_euler_discrete.py b/src/diffusers/schedulers/scheduling_euler_discrete.py index 6bd7e9b2a315b..2c397a818576b 100644 --- a/src/diffusers/schedulers/scheduling_euler_discrete.py +++ b/src/diffusers/schedulers/scheduling_euler_discrete.py @@ -503,7 +503,7 @@ def set_timesteps( if self.config.timestep_type == "continuous" and self.config.prediction_type == "v_prediction": self.timesteps = torch.Tensor([0.25 * sigma.log() for sigma in sigmas[:-1]]).to(device=device) elif self.config.use_flow_match: - self.timesteps = sigmas * self.config.num_train_timesteps + self.timesteps = sigmas[:-1] * self.config.num_train_timesteps else: self.timesteps = torch.from_numpy(timesteps.astype(np.float32)).to(device=device)