Skip to content

Commit b453004

Browse files
Youssef Adarrabyoussefadr
authored andcommitted
Fix failing CI on pytorch test on M1 (mps)
1 parent 6877850 commit b453004

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/schedulers/test_scheduler_heun.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -138,7 +138,7 @@ def test_full_loop_device_karras_sigmas(self):
138138
scheduler.set_timesteps(self.num_inference_steps, device=torch_device)
139139

140140
model = self.dummy_model()
141-
sample = self.dummy_sample_deter * scheduler.init_noise_sigma
141+
sample = self.dummy_sample_deter.to(torch_device) * scheduler.init_noise_sigma
142142
sample = sample.to(torch_device)
143143

144144
for t in scheduler.timesteps:

0 commit comments

Comments
 (0)