Update inference/flovd_demo.py

inference/flovd_demo.py

@@ -300,6 +300,8 @@ def patch_prepare_latents_safe():
 
         if latents is None:
             # noise = torch.randn_like(image_latents, generator=generator)
+            if generator.device != image_latents.device:
+                generator = generator.to(image_latents.device)
             noise = torch.randn(
                 image_latents.shape,
                 dtype=image_latents.dtype,