Spaces:
Running
on
Zero
Running
on
Zero
Update optimization.py
Browse files- optimization.py +2 -2
optimization.py
CHANGED
@@ -83,12 +83,12 @@ def optimize_pipeline_(pipeline: Callable[P, Any], *args: P.args, **kwargs: P.kw
|
|
83 |
# adapter_name="lightning_2", **kwargs_lora
|
84 |
# )
|
85 |
|
86 |
-
|
87 |
"Kijai/WanVideo_comfy",
|
88 |
weight_name="Stand-In/Stand-In_wan2.1_T2V_14B_ver1.0_fp16.safetensors",
|
89 |
#weight_name="Wan22-Lightning/Wan2.2-Lightning_T2V-A14B-4steps-lora_LOW_fp16.safetensors",
|
90 |
adapter_name="lightning_2", **kwargs_lora
|
91 |
-
|
92 |
|
93 |
|
94 |
|
|
|
83 |
# adapter_name="lightning_2", **kwargs_lora
|
84 |
# )
|
85 |
|
86 |
+
pipeline.load_lora_weights(
|
87 |
"Kijai/WanVideo_comfy",
|
88 |
weight_name="Stand-In/Stand-In_wan2.1_T2V_14B_ver1.0_fp16.safetensors",
|
89 |
#weight_name="Wan22-Lightning/Wan2.2-Lightning_T2V-A14B-4steps-lora_LOW_fp16.safetensors",
|
90 |
adapter_name="lightning_2", **kwargs_lora
|
91 |
+
)
|
92 |
|
93 |
|
94 |
|