Skip to content

Commit

Permalink
Update scheduling_euler_discrete.py
Browse files Browse the repository at this point in the history
If case syntax fix
  • Loading branch information
Senume authored Dec 5, 2023
1 parent a628f16 commit c22497a
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/diffusers/schedulers/scheduling_euler_discrete.py
Original file line number Diff line number Diff line change
Expand Up @@ -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)
Expand Down

0 comments on commit c22497a

Please sign in to comment.