Update pk_workflow.py

pk_workflow.py

@@ -210,7 +210,7 @@ def _euler_ancestral_step(scheduler, generator, model_output, timestep, sample,
         ratio = sigma_down / sigma
         prev_sample = ratio * x + (1 - ratio) * denoised
         if 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)
             prev_sample = (alpha_next / alpha_down) * prev_sample + noise * s_noise * renoise_coeff
 
     prev_sample = prev_sample.to(dtype=sample.dtype)