Spaces:
Running
on
Zero
Running
on
Zero
Update qwenimage/optimization.py
Browse files
qwenimage/optimization.py
CHANGED
|
@@ -76,7 +76,7 @@ def drain_module_parameters(module: torch.nn.Module):
|
|
| 76 |
|
| 77 |
|
| 78 |
# @ftimed
|
| 79 |
-
|
| 80 |
def optimize_pipeline_(
|
| 81 |
pipeline: Callable[P, Any],
|
| 82 |
cache_compiled=True,
|
|
|
|
| 76 |
|
| 77 |
|
| 78 |
# @ftimed
|
| 79 |
+
@spaces.GPU(duration=1500)
|
| 80 |
def optimize_pipeline_(
|
| 81 |
pipeline: Callable[P, Any],
|
| 82 |
cache_compiled=True,
|