enable zerogpu

app.py

@@ -261,7 +261,7 @@ def make_ref_cond(
 ):
     print("ready to run autoencoder")
     # print(f"image.device: {image.device}, type(image): {type(image)}")
-    image = image.to("cuda")
+    # image = image.to("cuda")
     latent = opts.latent_scaling_factor * autoencoder.encode(image[None, ...]).sample()
     return image[None, ...], latent
 

vqvae.py

@@ -49,7 +49,6 @@ class Autoencoder(nn.Module):
         # embedding space
         self.post_quant_conv = nn.Conv2d(emb_channels, z_channels, 1)
 
-    @spaces.GPU
     def encode(self, img: torch.Tensor) -> "GaussianDistribution":
         """
         ### Encode images to latent representation