Spaces:
Running
on
Zero
Running
on
Zero
Update app_lora_cpu.py
Browse files- app_lora_cpu.py +3 -3
app_lora_cpu.py
CHANGED
|
@@ -59,10 +59,10 @@ pipe.load_lora_weights(
|
|
| 59 |
kwargs_lora = {}
|
| 60 |
kwargs_lora["load_into_transformer_2"] = True
|
| 61 |
|
| 62 |
-
|
| 63 |
pipe.load_lora_weights(
|
| 64 |
"Kijai/WanVideo_comfy",
|
| 65 |
-
weight_name="
|
| 66 |
adapter_name="lightx2v_2", **kwargs_lora
|
| 67 |
)
|
| 68 |
|
|
@@ -140,7 +140,7 @@ pipe.load_lora_weights(
|
|
| 140 |
|
| 141 |
|
| 142 |
|
| 143 |
-
pipe.set_adapters(["lightx2v", "lightx2v_2", "high_noise_lora", "low_noise_lora","high_noise_lora1", "low_noise_lora1"], adapter_weights=[1
|
| 144 |
# 修改了lora_scale
|
| 145 |
pipe.fuse_lora(adapter_names=["lightx2v", "high_noise_lora","high_noise_lora1"], lora_scales=[3.0, 3.0,3.0], components=["transformer"])
|
| 146 |
# 修改了lora_scale
|
|
|
|
| 59 |
kwargs_lora = {}
|
| 60 |
kwargs_lora["load_into_transformer_2"] = True
|
| 61 |
|
| 62 |
+
#weight_name="Lightx2v/lightx2v_I2V_14B_480p_cfg_step_distill_rank128_bf16.safetensors",
|
| 63 |
pipe.load_lora_weights(
|
| 64 |
"Kijai/WanVideo_comfy",
|
| 65 |
+
weight_name="LoRAs/Wan22-Lightning/old/Wan2.2-Lightning_I2V-A14B-4steps-lora_LOW_fp16.safetensors",
|
| 66 |
adapter_name="lightx2v_2", **kwargs_lora
|
| 67 |
)
|
| 68 |
|
|
|
|
| 140 |
|
| 141 |
|
| 142 |
|
| 143 |
+
pipe.set_adapters(["lightx2v", "lightx2v_2", "high_noise_lora", "low_noise_lora","high_noise_lora1", "low_noise_lora1"], adapter_weights=[1.5, 1., 1., 1.,1.,1.])
|
| 144 |
# 修改了lora_scale
|
| 145 |
pipe.fuse_lora(adapter_names=["lightx2v", "high_noise_lora","high_noise_lora1"], lora_scales=[3.0, 3.0,3.0], components=["transformer"])
|
| 146 |
# 修改了lora_scale
|