Spaces:
Running
on
Zero
Running
on
Zero
Update optimization.py
Browse files- optimization.py +1 -1
optimization.py
CHANGED
@@ -40,7 +40,7 @@ INDUCTOR_CONFIGS = {
|
|
40 |
|
41 |
def optimize_pipeline_(pipeline: Callable[P, Any], *args: P.args, **kwargs: P.kwargs):
|
42 |
|
43 |
-
@spaces.GPU()
|
44 |
def compile_transformer():
|
45 |
|
46 |
# pipeline.load_lora_weights(
|
|
|
40 |
|
41 |
def optimize_pipeline_(pipeline: Callable[P, Any], *args: P.args, **kwargs: P.kwargs):
|
42 |
|
43 |
+
@spaces.GPU(duration=1200)
|
44 |
def compile_transformer():
|
45 |
|
46 |
# pipeline.load_lora_weights(
|