diff --git a/src/diffusers/schedulers/scheduling_euler_discrete.py b/src/diffusers/schedulers/scheduling_euler_discrete.py index 4e1b75845715..19ffca37c4e9 100644 --- a/src/diffusers/schedulers/scheduling_euler_discrete.py +++ b/src/diffusers/schedulers/scheduling_euler_discrete.py @@ -281,7 +281,7 @@ def set_timesteps(self, num_inference_steps: int, device: Union[str, torch.devic timesteps = np.array([self._sigma_to_t(sigma, log_sigmas) for sigma in sigmas]) # Checking 'sigmas' type before datatype conversion - if isinstance(sigmas, np.ndarray) + if isinstance(sigmas, np.ndarray): sigmas = torch.from_numpy(sigmas).to(dtype=torch.float32, device=device) else: sigmas = sigmas.to(dtype=torch.float32, device=device)