Update optimization.py

optimization.py

@@ -79,8 +79,16 @@ def optimize_pipeline_(pipeline: Callable[P, Any], *args: P.args, **kwargs: P.kw
             weight_name="Lightx2v/lightx2v_I2V_14B_480p_cfg_step_distill_rank128_bf16.safetensors", 
             adapter_name="lightx2v_2", **kwargs_lora
         )
-        pipeline.set_adapters(["lightx2v", "lightx2v_2"], adapter_weights=[1., 1.])
+        kwargs_lora_gh = {}
+        kwargs_lora_gh["load_into_transformer_2"] = True
+        pipe.load_lora_weights(
+            "GiorgioV/LoRA_for_WAN_22", 
+            weight_name="LoRA_GH.safetensors", 
+            adapter_name="lora_gh", **kwargs_lora_gh
+        )
+        pipeline.set_adapters(["lightx2v", "lightx2v_2", "lora_gh"], adapter_weights=[1., 1., 1.])
         pipeline.fuse_lora(adapter_names=["lightx2v"], lora_scale=3., components=["transformer"])
+        pipeline.fuse_lora(adapter_names=["lora_gh"], lora_scale=0.4, components=["transformer_2"])
         pipeline.fuse_lora(adapter_names=["lightx2v_2"], lora_scale=1., components=["transformer_2"])
         pipeline.unload_lora_weights()