lower ZeroGPU usage

demo.py

@@ -243,7 +243,7 @@ def preprocess_mask(mask):
 
 
 @spaces.GPU
-@torch.no_grad()
+@torch.no_grad(duration=3)
 def generate_latent_image(mask, class_selection, sampling_steps=50):
     """Generate a latent image based on mask, class selection, and sampling steps"""
 
@@ -307,7 +307,7 @@ def generate_latent_image(mask, class_selection, sampling_steps=50):
 
 
 @spaces.GPU
-@torch.no_grad()
+@torch.no_grad(duration=3)
 def decode_images(latents):
     """Decode latent representations to pixel space using a VAE.
 
@@ -386,7 +386,7 @@ def decode_latent_to_pixel(latent_image):
 
 
 @spaces.GPU
-@torch.no_grad()
+@torch.no_grad(duration=3)
 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)
@@ -413,7 +413,7 @@ def check_privacy(latent_image_numpy, class_selection):
 
 
 @spaces.GPU
-@torch.no_grad()
+@torch.no_grad(duration=3)
 def generate_animation(
     latent_image, ejection_fraction, sampling_steps=50, cfg_scale=1.0
 ):
@@ -502,7 +502,7 @@ def generate_animation(
 
 
 @spaces.GPU
-@torch.no_grad()
+@torch.no_grad(duration=3)
 def decode_animation(latent_animation):
     """Decode a latent animation to pixel space"""
     if latent_animation is None:
@@ -578,7 +578,7 @@ def convert_latent_to_display(latent_image):
 
 
 @spaces.GPU
-@torch.no_grad()
+@torch.no_grad(duration=3)
 def latent_animation_to_grayscale(latent_animation):
     """Convert multi-channel latent animation to grayscale for display"""
     if latent_animation is None: