enable zerogpu

app.py

@@ -328,7 +328,7 @@ def get_ref_anno(ref):
     ):
         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()
+        latent = opts.latent_scaling_factor * autoencoder.encode(image[None, ...].to(autoencoder.device)).sample()
         return image[None, ...], latent
 
     image_transform = Compose(