Update app.py

app.py

@@ -55,30 +55,34 @@ pipe = WanPipeline.from_pretrained(MODEL_ID,
 
 # ## 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(["lightning", "lightning_2", "high_noise_lora", "low_noise_lora"], adapter_weights=[1., 1., 1., 1.])
-# # 修改了lora_scale
-# pipe.fuse_lora(adapter_names=["lightning", "high_noise_lora"], lora_scales=[3.0, 3.0], components=["transformer"])
-# # 修改了lora_scale
-# pipe.fuse_lora(adapter_names=["lightning_2", "low_noise_lora"], lora_scales=[1.0, 1.0], components=["transformer_2"])
-# pipe.unload_lora_weights()
+pipe.load_lora_weights(
+    "wangkanai/wan22-fp8-loras-nsfw",
+    weight_name="loras/wan/wan22-action-doggystyle-t2v-14b-high.safetensors",
+    adapter_name="high_noise_lora",
+    token=os.environ.get("HF_TOKEN")
+)
+# 新增：加载你提供的low noise LoRA
+pipe.load_lora_weights(
+    "wangkanai/wan22-fp8-loras-nsfw",
+    weight_name="loras/wan/wan22-action-doggystyle-t2v-14b-low.safetensors",
+    adapter_name="low_noise_lora",
+    token=os.environ.get("HF_TOKEN"),
+    load_into_transformer_2=True
+)
+
+
+
+pipe.set_adapters(["lightning", "lightning_2", "high_noise_lora", "low_noise_lora"], adapter_weights=[1., 1., 1., 1.])
+# 修改了lora_scale
+pipe.fuse_lora(adapter_names=["lightning", "high_noise_lora"], lora_scales=[3.0, 3.0], components=["transformer"])
+# 修改了lora_scale
+pipe.fuse_lora(adapter_names=["lightning_2", "low_noise_lora"], lora_scales=[1.0, 1.0], components=["transformer_2"])
+pipe.unload_lora_weights()
 # ##
+# ## TEST LORA ENDS
+
+
+
 
 for i in range(3): 
     gc.collect()