Spaces:
Paused
Paused
reduce gpu duration
Browse files
app.py
CHANGED
|
@@ -44,7 +44,7 @@ def load_pipeline(accelerator, weight_dtype, args):
|
|
| 44 |
pipeline = pipeline.to(accelerator.device)
|
| 45 |
return pipeline
|
| 46 |
|
| 47 |
-
@spaces.GPU
|
| 48 |
def run(
|
| 49 |
instruction,
|
| 50 |
width_input,
|
|
|
|
| 44 |
pipeline = pipeline.to(accelerator.device)
|
| 45 |
return pipeline
|
| 46 |
|
| 47 |
+
@spaces.GPU
|
| 48 |
def run(
|
| 49 |
instruction,
|
| 50 |
width_input,
|