no@email.com commited on
Commit
fe250e7
·
1 Parent(s): 4b40584
Files changed (1) hide show
  1. app.py +9 -7
app.py CHANGED
@@ -61,8 +61,8 @@ pipe = WanImageToVideoPipeline.from_pretrained(
61
  # LOAD LORA ADAPTERS
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"
67
  )
68
  pipe.load_lora_weights(
@@ -71,8 +71,8 @@ pipe.load_lora_weights(
71
  adapter_name="i2v_scat"
72
  )
73
  pipe.load_lora_weights(
74
- "Kijai/WanVideo_comfy",
75
- weight_name="Lightx2v/lightx2v_I2V_14B_480p_cfg_step_distill_rank128_bf16.safetensors",
76
  adapter_name="lightx2v_2",
77
  load_into_transformer_2=True
78
  )
@@ -82,11 +82,13 @@ pipe.load_lora_weights(
82
  adapter_name="i2v_scat_2"
83
  )
84
 
85
- pipe.set_adapters(["lightx2v", "i2v_scat","lightx2v_2","i2v_scat_2"], adapter_weights=[1., 1., 1., 1.])
86
- pipe.fuse_lora(adapter_names=["lightx2v"], lora_scale=3., components=["transformer"])
87
  pipe.fuse_lora(adapter_names=["i2v_scat"], lora_scale=1., components=["transformer"])
88
- pipe.fuse_lora(adapter_names=["lightx2v_2"], lora_scale=1., components=["transformer_2"])
 
89
  pipe.fuse_lora(adapter_names=["i2v_scat_2"], lora_scale=1., components=["transformer_2"])
 
90
  pipe.unload_lora_weights()
91
 
92
  # =========================================================
 
61
  # LOAD LORA ADAPTERS
62
  # =========================================================
63
  pipe.load_lora_weights(
64
+ "lightx2v/Wan2.2-Lightning",
65
+ weight_name="Wan2.2-I2V-A14B-4steps-lora-rank64-Seko-V1/high_noise_model.safetensors",
66
  adapter_name="lightx2v"
67
  )
68
  pipe.load_lora_weights(
 
71
  adapter_name="i2v_scat"
72
  )
73
  pipe.load_lora_weights(
74
+ "lightx2v/Wan2.2-Lightning",
75
+ weight_name="Wan2.2-I2V-A14B-4steps-lora-rank64-Seko-V1/low_noise_model.safetensors",
76
  adapter_name="lightx2v_2",
77
  load_into_transformer_2=True
78
  )
 
82
  adapter_name="i2v_scat_2"
83
  )
84
 
85
+ pipe.set_adapters([ "i2v_scat","lightx2v","i2v_scat_2","lightx2v_2"], adapter_weights=[1., 1., 1., 1.])
86
+
87
  pipe.fuse_lora(adapter_names=["i2v_scat"], lora_scale=1., components=["transformer"])
88
+ pipe.fuse_lora(adapter_names=["lightx2v"], lora_scale=1., components=["transformer"])
89
+
90
  pipe.fuse_lora(adapter_names=["i2v_scat_2"], lora_scale=1., components=["transformer_2"])
91
+ pipe.fuse_lora(adapter_names=["lightx2v_2"], lora_scale=1., components=["transformer_2"])
92
  pipe.unload_lora_weights()
93
 
94
  # =========================================================