Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -47,7 +47,7 @@ pipe = load_models()
|
|
| 47 |
# return caption
|
| 48 |
|
| 49 |
|
| 50 |
-
@spaces.GPU(duration=
|
| 51 |
def generate_image(enhanced_caption, seed, num_inference_steps, guidance_scale):
|
| 52 |
# pipe.to("cuda")
|
| 53 |
pipe.enable_model_cpu_offload()
|
|
|
|
| 47 |
# return caption
|
| 48 |
|
| 49 |
|
| 50 |
+
@spaces.GPU(duration=40)
|
| 51 |
def generate_image(enhanced_caption, seed, num_inference_steps, guidance_scale):
|
| 52 |
# pipe.to("cuda")
|
| 53 |
pipe.enable_model_cpu_offload()
|