Spaces:
Paused
Paused
rizavelioglu
commited on
Commit
•
80cc34d
1
Parent(s):
f32ebdc
reduce zero-gpu duration
Browse files
app.py
CHANGED
@@ -78,7 +78,7 @@ torch.cuda.empty_cache()
|
|
78 |
|
79 |
|
80 |
# Define image generation function
|
81 |
-
@spaces.GPU(duration=
|
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
|