Update app.py

app.py

@@ -53,32 +53,32 @@ 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()
-##
+# ## 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()
+# ##
 
 for i in range(3): 
     gc.collect()