Update app_lora_cpu.py

app_lora_cpu.py

@@ -57,55 +57,112 @@ pipe.load_lora_weights(
 )
 kwargs_lora = {}
 kwargs_lora["load_into_transformer_2"] = True
-
-
 pipe.load_lora_weights(
     "Kijai/WanVideo_comfy",
     weight_name="Lightx2v/lightx2v_I2V_14B_480p_cfg_step_distill_rank128_bf16.safetensors",
     adapter_name="lightx2v_2", **kwargs_lora
 )
+# 原始：加载 v8nomal high noise LoRA
+pipe.load_lora_weights(
+    "mimigona1/twote2",
+    weight_name="ultrade_high.safetensors",
+    adapter_name="high_noise_lora",
+    token=os.environ.get("HF_TOKEN")
+)
+# 原始：加载 v8nomal low noise LoRA
+pipe.load_lora_weights(
+    "mimigona1/twote2",
+    weight_name="ultrade_low.safetensors",
+    adapter_name="low_noise_lora",
+    token=os.environ.get("HF_TOKEN"),
+    load_into_transformer_2=True
+)
 
+# 新增：加载 high noise LoRA (dremal)
+pipe.load_lora_weights(
+    "mimigona1/twote2",
+    weight_name="pusylick_high.safetensors",
+    adapter_name="high_dremal_lora",
+    token=os.environ.get("HF_TOKEN")
+)
+# 新增：加载 low noise LoRA (dremal)
+pipe.load_lora_weights(
+    "mimigona1/twote2",
+    weight_name="pusylick_low.safetensors",
+    adapter_name="low_dremal_lora",
+    token=os.environ.get("HF_TOKEN"),
+    load_into_transformer_2=True
+)
 
+# 🚀 增加额外新 LoRA 1: missimd_high.safetensors (high noise)
+pipe.load_lora_weights(
+    "mimigona1/twote2",
+    weight_name="redouble_high.safetensors",
+    adapter_name="high_missimd_lora", # 新增 adapter name
+    token=os.environ.get("HF_TOKEN")
+)
+# 🚀 增加额外新 LoRA 1: missimd_low.safetensors (low noise)
+pipe.load_lora_weights(
+    "mimigona1/twote2",
+    weight_name="redouble_low.safetensors",
+    adapter_name="low_missimd_lora", # 新增 adapter name
+    token=os.environ.get("HF_TOKEN"),
+    load_into_transformer_2=True
+)
 
-# 新增：加载你提供的high noise LoRA
-
+# 🚀 增加额外新 LoRA 2: sitfov_high.safetensors (high noise) - 配置与 dremal 一致
 pipe.load_lora_weights(
-    "rahul7star/wan2.2Lora",
-    weight_name="DR34ML4Y_I2V_14B_HIGH.safetensors",
-    adapter_name="high_noise_lora",
+    "mimigona1/twote2",
+    weight_name="dog02_high.safetensors",
+    adapter_name="high_ultrade_lora", # 适配器名称
     token=os.environ.get("HF_TOKEN")
 )
-# 新增：加载你提供的low noise LoRA
+# 🚀 增加额外新 LoRA 2: sitfov_low.safetensors (low noise) - 配置与 dremal 一致
 pipe.load_lora_weights(
-    "rahul7star/wan2.2Lora",
-    weight_name="DR34ML4Y_I2V_14B_LOW.safetensors",
-    adapter_name="low_noise_lora",
+    "mimigona1/twote2",
+    weight_name="dog02_low.safetensors",
+    adapter_name="low_ultrade_lora", # 适配器名称
     token=os.environ.get("HF_TOKEN"),
     load_into_transformer_2=True
 )
 
 
-## thi s still gpood 
-# pipe.load_lora_weights(
-#     "rahul7star/wan2.2Lora",
-#     weight_name="wan2.2_i2v_highnoise_pov_missionary_v1.0.safetensors",
-#     adapter_name="high_noise_lora",
-#     token=os.environ.get("HF_TOKEN")
-# )
-# # 新增：加载你提供的low noise LoRA
-# pipe.load_lora_weights(
-#     "rahul7star/wan2.2Lora",
-#     weight_name="wan2.2_i2v_lownoise_pov_missionary_v1.0.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"])
+# 修改 set_adapters: 包含所有 LoRA
+pipe.set_adapters(
+    [
+        "lightx2v", "lightx2v_2", 
+        "high_noise_lora", "low_noise_lora", 
+        "high_dremal_lora", "low_dremal_lora",
+        "high_missimd_lora", "low_missimd_lora", # 🚀 新增 missimd LoRA
+        "high_ultrade_lora", "low_ultrade_lora" # 🚀 新增 ultrade LoRA
+    ],
+    # 权重列表对应上述适配器。missimd 和 ultrade 均设置为 1.0
+    adapter_weights=[1.5, 1.2, 0,0 , 0.7, 0.7, 0.7, 0.7, 0.7, 0.7] 
+)
+# 修改 fuse_lora (transformer): 加入 high_missimd_lora, high_ultrade_lora (lora_scale=2.0)
+pipe.fuse_lora(
+    adapter_names=[
+        "lightx2v", 
+        "high_noise_lora", 
+        "high_dremal_lora", 
+        "high_missimd_lora", # 🚀 新增
+        "high_ultrade_lora" # 🚀 新增
+    ], 
+    lora_scales=[4.0, 3.0, 2.0, 2.0, 2.0], # 🚀 missimd 和 ultrade scale 设置为 2.0
+    components=["transformer"]
+)
+# 修改 fuse_lora (transformer_2): 加入 low_missimd_lora, low_ultrade_lora (lora_scale=1.0)
+pipe.fuse_lora(
+    adapter_names=[
+        "lightx2v_2", 
+        "low_noise_lora", 
+        "low_dremal_lora", 
+        "low_missimd_lora", # 🚀 新增
+        "low_ultrade_lora" # 🚀 新增
+    ], 
+    lora_scales=[2.0, 1.5, 1.0, 1.0, 1.0], # 🚀 missimd 和 ultrade scale 设置为 1.0
+    components=["transformer_2"]
+)
 pipe.unload_lora_weights()
 
 quantize_(pipe.text_encoder, Int8WeightOnlyConfig())