rahul7star commited on
Commit
f4ca778
·
verified ·
1 Parent(s): 5d16c30

Update app_lora_cpu.py

Browse files
Files changed (1) hide show
  1. 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="Lightx2v/lightx2v_I2V_14B_480p_cfg_step_distill_rank128_bf16.safetensors",
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., 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
 
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