Update app.py

app.py

@@ -61,33 +61,42 @@ pipe = WanImageToVideoPipeline.from_pretrained(MODEL_ID,
 ).to('cuda')
 
 pipe.load_lora_weights(
-    "obsxrver/wan2.2-i2v-scat",
-    weight_name="WAN2.2-I2V-HighNoise_scat-xxi-i2v.safetensors",
-    adapter_name="i2v_scat"
+    "Kijai/WanVideo_comfy", 
+    weight_name="Lightx2v/lightx2v_I2V_14B_480p_cfg_step_distill_rank128_bf16.safetensors", 
+    adapter_name="lightx2v"
 )
+
+kwargs_lora_h = {}
+kwargs_lora_h["load_into_transformer"] = True
 pipe.load_lora_weights(
-    "lightx2v/Wan2.2-Lightning",
-    weight_name="Wan2.2-I2V-A14B-4steps-lora-rank64-Seko-V1/high_noise_model.safetensors",
-    adapter_name="lightx2v"
+    "GiorgioV/LoRA_for_WAN_22", 
+    weight_name="I2V_14B_HIGH.safetensors", 
+    adapter_name="lora_h", **kwargs_lora_h
 )
 
+
+kwargs_lora = {}
+kwargs_lora["load_into_transformer_2"] = True
 pipe.load_lora_weights(
-    "obsxrver/wan2.2-i2v-scat",
-    weight_name="WAN2.2-I2V-LowNoise_scat-xxi-i2v.safetensors",
-    adapter_name="i2v_scat_2",
-    load_into_transformer_2=True
+    "Kijai/WanVideo_comfy", 
+    weight_name="Lightx2v/lightx2v_I2V_14B_480p_cfg_step_distill_rank128_bf16.safetensors", 
+    adapter_name="lightx2v_2", **kwargs_lora
 )
+
+kwargs_lora_l = {}
+kwargs_lora_l["load_into_transformer_2"] = True
 pipe.load_lora_weights(
-    "lightx2v/Wan2.2-Lightning",
-    weight_name="Wan2.2-I2V-A14B-4steps-lora-rank64-Seko-V1/low_noise_model.safetensors",
-    adapter_name="lightx2v_2",
-    load_into_transformer_2=True
+    "GiorgioV/LoRA_for_WAN_22", 
+    weight_name="I2V_14B_LOW.safetensors", 
+    adapter_name="lora_l", **kwargs_lora_l
 )
 
 
-pipe.set_adapters([ "i2v_scat","lightx2v","i2v_scat_2","lightx2v_2"], adapter_weights=[0.9, 1.0, 0.95, 1.0])
-pipe.fuse_lora(adapter_names=["i2v_scat","lightx2v"], lora_scale=1., components=["transformer"])
-pipe.fuse_lora(adapter_names=["i2v_scat_2","lightx2v_2"], lora_scale=1., components=["transformer_2"])
+pipe.set_adapters(["lightx2v", "lora_h", "lora_gh", "lightx2v_2", "lora_l", "lora_gl"], adapter_weights=[1., 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.4, components=["transformer"])
+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.unload_lora_weights()
 
 quantize_(pipe.text_encoder, Int8WeightOnlyConfig())