zeroGPU

demo.py

@@ -30,6 +30,8 @@ torch.set_grad_enabled(False)
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 dtype = torch.float32
 
+print(f"Using device: {device}")
+
 # 4f4 latent space
 B, T, C, H, W = 1, 64, 4, 28, 28
 
@@ -141,7 +143,7 @@ def get_vae_scaler(path):
     return scaler
 
 
-generator = torch.Generator(device=device).manual_seed(0)
+# generator = torch.Generator(device=device).manual_seed(0)
 
 lifm = load_model("https://huggingface.co/HReynaud/EchoFlow/tree/main/lifm/FMiT-S2-4f4")
 lifm = lifm.to(device, dtype=dtype)
@@ -271,7 +273,7 @@ def generate_latent_image(mask, class_selection, sampling_steps=50):
         (B, C, H, W),
         device=device,
         dtype=dtype,
-        generator=generator,
+        # generator=generator,
     )
 
     lifm.forward_original = lifm.forward
@@ -439,7 +441,7 @@ def generate_animation(
         (B, C, T, H, W),
         device=device,
         dtype=dtype,
-        generator=generator,
+        # generator=generator,
     )
 
     # print(