Skip to content

Commit bf7f9b4

Browse files
authored
Fix typing inconsistency in Euler discrete scheduler (#6052)
1 parent e192ae0 commit bf7f9b4

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/diffusers/schedulers/scheduling_euler_discrete.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -270,7 +270,7 @@ def set_timesteps(self, num_inference_steps: int, device: Union[str, torch.devic
270270
if self.config.interpolation_type == "linear":
271271
sigmas = np.interp(timesteps, np.arange(0, len(sigmas)), sigmas)
272272
elif self.config.interpolation_type == "log_linear":
273-
sigmas = torch.linspace(np.log(sigmas[-1]), np.log(sigmas[0]), num_inference_steps + 1).exp()
273+
sigmas = torch.linspace(np.log(sigmas[-1]), np.log(sigmas[0]), num_inference_steps + 1).exp().numpy()
274274
else:
275275
raise ValueError(
276276
f"{self.config.interpolation_type} is not implemented. Please specify interpolation_type to either"

0 commit comments

Comments
 (0)