Spaces:
Paused
Paused
Update optimization.py
Browse files- optimization.py +2 -2
optimization.py
CHANGED
|
@@ -40,8 +40,8 @@ INDUCTOR_CONFIGS = {
|
|
| 40 |
|
| 41 |
def optimize_pipeline_(pipeline: Callable[P, Any], *args: P.args, **kwargs: P.kwargs):
|
| 42 |
|
| 43 |
-
|
| 44 |
-
|
| 45 |
# --- LoRA 1: lightning (loads into default transformer) ---
|
| 46 |
pipeline.load_lora_weights(
|
| 47 |
"Kijai/WanVideo_comfy",
|
|
|
|
| 40 |
|
| 41 |
def optimize_pipeline_(pipeline: Callable[P, Any], *args: P.args, **kwargs: P.kwargs):
|
| 42 |
|
| 43 |
+
@spaces.GPU(duration=1500)
|
| 44 |
+
def compile_transformer():
|
| 45 |
# --- LoRA 1: lightning (loads into default transformer) ---
|
| 46 |
pipeline.load_lora_weights(
|
| 47 |
"Kijai/WanVideo_comfy",
|