We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 1225ab2 commit c56cd09Copy full SHA for c56cd09
tests/schedulers/test_scheduler_heun.py
@@ -138,7 +138,7 @@ def test_full_loop_device_karras_sigmas(self):
138
scheduler.set_timesteps(self.num_inference_steps, device=torch_device)
139
140
model = self.dummy_model()
141
- sample = self.dummy_sample_deter * scheduler.init_noise_sigma
+ sample = self.dummy_sample_deter.to(torch_device) * scheduler.init_noise_sigma
142
sample = sample.to(torch_device)
143
144
for t in scheduler.timesteps:
0 commit comments