Update app.py

app.py

@@ -52,6 +52,34 @@ pipe = WanPipeline.from_pretrained(MODEL_ID,
 ).to('cuda')
 
 
+
+## TEST LORA
+
+pipe.load_lora_weights(
+    "rahul7star/wan2.2Lora",
+    weight_name="DR34ML4Y_I2V_14B_HIGH.safetensors",
+    adapter_name="high_noise_lora",
+    token=os.environ.get("HF_TOKEN")
+)
+# 新增：加载你提供的low noise LoRA
+pipe.load_lora_weights(
+    "rahul7star/wan2.2Lora",
+    weight_name="DR34ML4Y_I2V_14B_LOW.safetensors",
+    adapter_name="low_noise_lora",
+    token=os.environ.get("HF_TOKEN"),
+    load_into_transformer_2=True
+)
+
+
+
+pipe.set_adapters(["lightx2v", "lightx2v_2", "high_noise_lora", "low_noise_lora"], adapter_weights=[1., 1., 1., 1.])
+# 修改了lora_scale
+pipe.fuse_lora(adapter_names=["lightx2v", "high_noise_lora"], lora_scales=[3.0, 3.0], components=["transformer"])
+# 修改了lora_scale
+pipe.fuse_lora(adapter_names=["lightx2v_2", "low_noise_lora"], lora_scales=[1.0, 1.0], components=["transformer_2"])
+pipe.unload_lora_weights()
+##
+
 for i in range(3): 
     gc.collect()
     torch.cuda.synchronize()