From c22497a6d6781bf65fab5073c420784b6688076a Mon Sep 17 00:00:00 2001 From: Thushit Kumar R <101030274+Senume@users.noreply.github.com> Date: Tue, 5 Dec 2023 14:46:18 +0530 Subject: [PATCH] Update scheduling_euler_discrete.py If case syntax fix --- 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 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)