Spaces:
Running
on
Zero
Running
on
Zero
Update optimization.py
Browse files- optimization.py +1 -1
optimization.py
CHANGED
|
@@ -81,7 +81,7 @@ def optimize_pipeline_(pipeline: Callable[P, Any], *args: P.args, **kwargs: P.kw
|
|
| 81 |
)
|
| 82 |
kwargs_lora_gh = {}
|
| 83 |
kwargs_lora_gh["load_into_transformer_2"] = True
|
| 84 |
-
|
| 85 |
"GiorgioV/LoRA_for_WAN_22",
|
| 86 |
weight_name="LoRA_GH.safetensors",
|
| 87 |
adapter_name="lora_gh", **kwargs_lora_gh
|
|
|
|
| 81 |
)
|
| 82 |
kwargs_lora_gh = {}
|
| 83 |
kwargs_lora_gh["load_into_transformer_2"] = True
|
| 84 |
+
pipeline.load_lora_weights(
|
| 85 |
"GiorgioV/LoRA_for_WAN_22",
|
| 86 |
weight_name="LoRA_GH.safetensors",
|
| 87 |
adapter_name="lora_gh", **kwargs_lora_gh
|