Update h3_dpmpp_2s_ancestral.py

h3_dpmpp_2s_ancestral.py

@@ -70,7 +70,7 @@ def _dpmpp_2s_ancestral_combine(
     sigma_down_i_ratio = sigma_down / sigma_i
     x = sigma_down_i_ratio * x + (1 - sigma_down_i_ratio) * d_i
     if float(sigma_ip1) > 0 and eta > 0:
-        noise = torch.randn(x.shape, dtype=x.dtype, device=x.device, generator=generator)
+        noise = torch.randn(x.shape, dtype=x.dtype, device="cpu", generator=generator).to(x.device)
         x = (alpha_ip1 / alpha_down) * x + noise * s_noise * renoise_coeff
     return x