Spaces:
Running
on
Zero
Running
on
Zero
fix:gpu duration too large
Browse files
app.py
CHANGED
|
@@ -85,7 +85,7 @@ def load_pipeline(accelerator, weight_dtype, args):
|
|
| 85 |
pipeline = pipeline.to(accelerator.device)
|
| 86 |
return pipeline
|
| 87 |
|
| 88 |
-
@spaces.GPU(duration=
|
| 89 |
def run(
|
| 90 |
instruction,
|
| 91 |
width_input,
|
|
|
|
| 85 |
pipeline = pipeline.to(accelerator.device)
|
| 86 |
return pipeline
|
| 87 |
|
| 88 |
+
@spaces.GPU(duration=120)
|
| 89 |
def run(
|
| 90 |
instruction,
|
| 91 |
width_input,
|