Update inference/flovd_demo.py

inference/flovd_demo.py

@@ -299,7 +299,14 @@ def patch_prepare_latents_safe():
             image_latents = torch.cat([image_latents, latent_padding], dim=2)
 
         if latents is None:
-            noise = torch.randn_like(image_latents, generator=generator)
+            # noise = torch.randn_like(image_latents, generator=generator)
+            noise = torch.randn(
+                image_latents.shape,
+                dtype=image_latents.dtype,
+                device=image_latents.device,
+                generator=generator
+            )
+
             latents = noise.to(device=device, dtype=dtype)
 
         return latents, image_latents.to(device, dtype=dtype)