GiorgioV commited on
Commit
c229e0c
·
verified ·
1 Parent(s): 0e846c6

Update optimization.py

Browse files
Files changed (1) hide show
  1. 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
- pipe.load_lora_weights(
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