Spaces:
Running
on
Zero
Running
on
Zero
Update optimization.py
Browse files- optimization.py +1 -1
optimization.py
CHANGED
@@ -74,7 +74,7 @@ def optimize_pipeline_(pipeline: Callable[P, Any], *args: P.args, **kwargs: P.kw
|
|
74 |
kwargs_lora["load_into_transformer_2"] = True
|
75 |
pipeline.load_lora_weights(
|
76 |
"Kijai/WanVideo_comfy",
|
77 |
-
weight_name="
|
78 |
#weight_name="Wan22-Lightning/Wan2.2-Lightning_T2V-A14B-4steps-lora_LOW_fp16.safetensors",
|
79 |
adapter_name="lightning_2", **kwargs_lora
|
80 |
)
|
|
|
74 |
kwargs_lora["load_into_transformer_2"] = True
|
75 |
pipeline.load_lora_weights(
|
76 |
"Kijai/WanVideo_comfy",
|
77 |
+
weight_name="Pusa/Wan21_PusaV1_LoRA_14B_rank512_bf16.safetensors",
|
78 |
#weight_name="Wan22-Lightning/Wan2.2-Lightning_T2V-A14B-4steps-lora_LOW_fp16.safetensors",
|
79 |
adapter_name="lightning_2", **kwargs_lora
|
80 |
)
|