enable zerogpu

app.py

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