Update app.py

app.py

@@ -70,36 +70,28 @@ pipe.load_lora_weights(
 
 # 新增：加载你提供的high noise LoRA
 
-# pipe.load_lora_weights(
-#     "GiorgioV/LoRA_for_WAN_22",
-#     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(
-#     "GiorgioV/LoRA_for_WAN_22",
-#     weight_name="DR34ML4Y_I2V_14B_LOW.safetensors",
-#     adapter_name="low_noise_lora",
-#     token=os.environ.get("HF_TOKEN"),
-#     load_into_transformer_2=True
-# )
-
 pipe.load_lora_weights(
-    "profpeng/nsfwv2",
-    weight_name="DR34ML4Y_I2V_14B_HIGH_V2.safetensors",
+    "GiorgioV/LoRA_for_WAN_22",
+    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(
-    "profpeng/nsfwv2",
-    weight_name="DR34ML4Y_I2V_14B_LOW_V2.safetensors",
+    "GiorgioV/LoRA_for_WAN_22",
+    weight_name="DR34ML4Y_I2V_14B_LOW.safetensors",
     adapter_name="low_noise_lora",
     token=os.environ.get("HF_TOKEN"),
     load_into_transformer_2=True
 )
 
+pipe.load_lora_weights(
+    "profpeng/wanhardcut",
+    weight_name="hard_cut_200_wan_i2v_high.safetensors",
+    adapter_name="high_noise_loraHC",
+    token=os.environ.get("HF_TOKEN")
+)
+
 
 pipe.load_lora_weights(
     "profpeng/wanlegupcutscenesex",
@@ -118,10 +110,11 @@ pipe.load_lora_weights(
 
 
 pipe.set_adapters(["lightx2v", "lightx2v_2",
+                   "high_noise_loraHC",
                    "high_noise_lora", "low_noise_lora",
-                   "high_noise_lora1", "low_noise_lora1"], adapter_weights=[1.5, 1., 1., 1., 1., 1.])
+                   "high_noise_lora1", "low_noise_lora1"], adapter_weights=[1.5, 1., 1., 1., 1., 1., 1.])
 # 修改了lora_scale
-pipe.fuse_lora(adapter_names=["lightx2v", "high_noise_lora", "high_noise_lora1"], lora_scales=[3.0, 3.0, 3.0], components=["transformer"])
+pipe.fuse_lora(adapter_names=["lightx2v", "high_noise_lora", "high_noise_loraHC", "high_noise_lora1"], lora_scales=[3.0, 3.0, 3.0, 3.0], components=["transformer"])
 # 修改了lora_scale
 pipe.fuse_lora(adapter_names=["lightx2v_2", "low_noise_lora", "low_noise_lora1"], lora_scales=[1.0, 1.0, 1.0], components=["transformer_2"])