From bd4e9780be0782b7e37f678bb212157e2b531f29 Mon Sep 17 00:00:00 2001 From: Ian Baldwin Date: Mon, 4 Dec 2023 19:53:19 -0800 Subject: [PATCH] Fix typing inconsistency in Euler discrete scheduler --- 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 53dc2ae15432..e33332d0a9d4 100644 --- a/src/diffusers/schedulers/scheduling_euler_discrete.py +++ b/src/diffusers/schedulers/scheduling_euler_discrete.py @@ -269,7 +269,7 @@ def set_timesteps(self, num_inference_steps: int, device: Union[str, torch.devic if self.config.interpolation_type == "linear": sigmas = np.interp(timesteps, np.arange(0, len(sigmas)), sigmas) elif self.config.interpolation_type == "log_linear": - sigmas = torch.linspace(np.log(sigmas[-1]), np.log(sigmas[0]), num_inference_steps + 1).exp() + sigmas = torch.linspace(np.log(sigmas[-1]), np.log(sigmas[0]), num_inference_steps + 1).exp().numpy() else: raise ValueError( f"{self.config.interpolation_type} is not implemented. Please specify interpolation_type to either"