Update app.py

app.py

@@ -66,15 +66,13 @@ pipe.load_lora_weights(
     adapter_name="lightx2v"
 )
 
-kwargs_lora_h = {}
 kwargs_lora_h["load_into_transformer"] = True
 pipe.load_lora_weights(
     "GiorgioV/LoRA_for_WAN_22", 
-    weight_name="P003-SD-i2v-v10-000010_converted.safetensors", 
+    weight_name="LoRA_exp.safetensors", 
     adapter_name="lora_h", **kwargs_lora_h
 )
 
-
 kwargs_lora = {}
 kwargs_lora["load_into_transformer_2"] = True
 pipe.load_lora_weights(
@@ -83,20 +81,27 @@ pipe.load_lora_weights(
     adapter_name="lightx2v_2", **kwargs_lora
 )
 
-kwargs_lora_l = {}
 kwargs_lora_l["load_into_transformer_2"] = True
 pipe.load_lora_weights(
     "GiorgioV/LoRA_for_WAN_22", 
-    weight_name="LoRA_FH.safetensors", 
+    weight_name="LoRA_exp_L.safetensors", 
     adapter_name="lora_l", **kwargs_lora_l
 )
+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
+)
 
 
-pipe.set_adapters(["lightx2v", "lora_h", "lightx2v_2", "lora_l"], adapter_weights=[1., 1., 1., 1.])
+pipe.set_adapters(["lightx2v", "lora_h", "lightx2v_2", "lora_l", "lora_gh"], adapter_weights=[1., 1., 1., 1., 1.])
 pipe.fuse_lora(adapter_names=["lightx2v"], lora_scale=3., components=["transformer"])
 pipe.fuse_lora(adapter_names=["lora_h"], lora_scale=0.1, components=["transformer"])
-pipe.fuse_lora(adapter_names=["lightx2v_2"], lora_scale=1.0, components=["transformer_2"])
-pipe.fuse_lora(adapter_names=["lora_l"], lora_scale=0.8, components=["transformer_2"])
+pipe.fuse_lora(adapter_names=["lightx2v_2"], lora_scale=1., components=["transformer_2"])
+pipe.fuse_lora(adapter_names=["lora_l"], lora_scale=1., components=["transformer_2"])
+pipe.fuse_lora(adapter_names=["lora_gh"], lora_scale=0.6, components=["transformer_2"])
 pipe.unload_lora_weights()
 
 quantize_(pipe.text_encoder, Int8WeightOnlyConfig())