lower ZeroGPU usage

demo.py

@@ -242,8 +242,8 @@ def preprocess_mask(mask):
     return np.array(mask_pil)
 
 
-@spaces.GPU
-@torch.no_grad(duration=3)
+@spaces.GPU(duration=3)
+@torch.no_grad()
 def generate_latent_image(mask, class_selection, sampling_steps=50):
     """Generate a latent image based on mask, class selection, and sampling steps"""
 
@@ -306,8 +306,8 @@ def generate_latent_image(mask, class_selection, sampling_steps=50):
     return latent_image  # B x C x H x W
 
 
-@spaces.GPU
-@torch.no_grad(duration=3)
+@spaces.GPU(duration=3)
+@torch.no_grad()
 def decode_images(latents):
     """Decode latent representations to pixel space using a VAE.
 
@@ -385,8 +385,8 @@ def decode_latent_to_pixel(latent_image):
     return decoded_image
 
 
-@spaces.GPU
-@torch.no_grad(duration=3)
+@spaces.GPU(duration=3)
+@torch.no_grad()
 def check_privacy(latent_image_numpy, class_selection):
     """Check if the latent image is too similar to database images"""
     latent_image = torch.from_numpy(latent_image_numpy).to(device, dtype=dtype)
@@ -412,8 +412,8 @@ def check_privacy(latent_image_numpy, class_selection):
         )
 
 
-@spaces.GPU
-@torch.no_grad(duration=3)
+@spaces.GPU(duration=3)
+@torch.no_grad()
 def generate_animation(
     latent_image, ejection_fraction, sampling_steps=50, cfg_scale=1.0
 ):
@@ -501,8 +501,8 @@ def generate_animation(
     return synthetic_video.detach().cpu()  # B x C x T x H x W
 
 
-@spaces.GPU
-@torch.no_grad(duration=3)
+@spaces.GPU(duration=3)
+@torch.no_grad()
 def decode_animation(latent_animation):
     """Decode a latent animation to pixel space"""
     if latent_animation is None:
@@ -577,8 +577,8 @@ def convert_latent_to_display(latent_image):
     return display_image
 
 
-@spaces.GPU
-@torch.no_grad(duration=3)
+@spaces.GPU(duration=3)
+@torch.no_grad()
 def latent_animation_to_grayscale(latent_animation):
     """Convert multi-channel latent animation to grayscale for display"""
     if latent_animation is None: