enable zerogpu

app.py

@@ -327,7 +327,6 @@ def get_ref_anno(ref):
         image
     ):
         print("ready to run autoencoder")
-        image = image.cuda()
         latent = opts.latent_scaling_factor * autoencoder.encode(image[None, ...]).sample()
         return image[None, ...], latent
 
@@ -338,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)
+    image = image_transform(img).to(device)
     kpts_valid = check_keypoints_validity(keypts, opts.image_size)
     heatmaps = torch.tensor(
         keypoint_heatmap(