enable zerogpu

app.py

@@ -505,15 +505,16 @@ def reset_kps(img, keypoints, side: Literal["right", "left"]):
         keypoints[1] = []
     return img, keypoints
 
-@spaces.GPU(duration=120)
+@spaces.GPU(duration=60)
 def sample_diff(ref_cond, target_cond, target_keypts, num_gen, seed, cfg):
     set_seed(seed)
     z = torch.randn(
         (num_gen, opts.latent_dim, opts.latent_size[0], opts.latent_size[1]),
         device=device,
     )
-    target_cond = target_cond.repeat(num_gen, 1, 1, 1)
-    ref_cond = ref_cond.repeat(num_gen, 1, 1, 1)
+    print(f"z.device: {z.device}")
+    target_cond = target_cond.repeat(num_gen, 1, 1, 1).to(z.device)
+    ref_cond = ref_cond.repeat(num_gen, 1, 1, 1).to(z.device)
     # novel view synthesis mode = off
     nvs = torch.zeros(num_gen, dtype=torch.int, device=device)
     z = torch.cat([z, z], 0)