enable zerogpu

app.py

@@ -337,7 +337,7 @@ def get_ref_anno(ref):
             Normalize(mean=[0.5, 0.5, 0.5], std=[0.5, 0.5, 0.5], inplace=True),
         ]
     )
-    image = image_transform(img).to(device)
+    image = image_transform(img)
     kpts_valid = check_keypoints_validity(keypts, opts.image_size)
     heatmaps = torch.tensor(
         keypoint_heatmap(
@@ -345,7 +345,7 @@ def get_ref_anno(ref):
         )
         * kpts_valid[:, None, None],
         dtype=torch.float,
-        device=device,
+        # device=device,
     )[None, ...]
     mask = torch.tensor(
         cv2.resize(
@@ -354,7 +354,7 @@ def get_ref_anno(ref):
             interpolation=cv2.INTER_NEAREST,
         ),
         dtype=torch.float,
-        device=device,
+        # device=device,
     ).unsqueeze(0)[None, ...]
     image, latent = make_ref_cond(
         image,