enable zerogpu

app.py

@@ -262,6 +262,9 @@ def make_ref_cond(
 ):
     print("ready to run autoencoder")
     # print(f"image.device: {image.device}, type(image): {type(image)}")
+    os.environ["CUDA_VISIBLE_DEVICES"] = "0"
+    torch.cuda.set_device(0)
+    image = image.to("cuda")
     latent = opts.latent_scaling_factor * autoencoder.encode(image[None, ...]).sample()
     return image[None, ...], latent
 
@@ -338,9 +341,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)
-    image = image.numpy()
-    # image = torch.tensor(image)
+    image = image_transform(img).to(device)
     kpts_valid = check_keypoints_validity(keypts, opts.image_size)
     heatmaps = torch.tensor(
         keypoint_heatmap(