rizavelioglu commited on
Commit
80cc34d
1 Parent(s): f32ebdc

reduce zero-gpu duration

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -78,7 +78,7 @@ torch.cuda.empty_cache()
78
 
79
 
80
  # Define image generation function
81
- @spaces.GPU(duration=75)
82
  @torch.no_grad()
83
  def generate_image(input_image, seed=42, guidance_scale=2.0, num_inference_steps=50, is_upscale=False):
84
  # Configure scheduler
 
78
 
79
 
80
  # Define image generation function
81
+ @spaces.GPU(duration=10)
82
  @torch.no_grad()
83
  def generate_image(input_image, seed=42, guidance_scale=2.0, num_inference_steps=50, is_upscale=False):
84
  # Configure scheduler