GiorgioV commited on
Commit
7f941d7
·
verified ·
1 Parent(s): b799890

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +18 -25
app.py CHANGED
@@ -61,40 +61,33 @@ pipe = WanImageToVideoPipeline.from_pretrained(MODEL_ID,
61
  ).to('cuda')
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
-
69
- kwargs_lora_h = {}
70
- kwargs_lora_h["load_into_transformer"] = True
71
  pipe.load_lora_weights(
72
- "GiorgioV/LoRA_for_WAN_22",
73
- weight_name="I2V_14B_HIGH.safetensors",
74
- adapter_name="lora_h", **kwargs_lora_h
75
  )
76
 
77
- kwargs_lora = {}
78
- kwargs_lora["load_into_transformer_2"] = True
79
  pipe.load_lora_weights(
80
- "lightx2v/Wan2.2-Lightning",
81
- weight_name="Wan2.2-I2V-A14B-4steps-lora-rank64-Seko-V1/low_noise_model.safetensors",
82
- adapter_name="lightx2v_2", **kwargs_lora
 
83
  )
84
-
85
- kwargs_lora_l = {}
86
- kwargs_lora_l["load_into_transformer_2"] = True
87
  pipe.load_lora_weights(
88
- "GiorgioV/LoRA_for_WAN_22",
89
- weight_name="I2V_14B_LOW.safetensors",
90
- adapter_name="lora_l", **kwargs_lora_l
 
91
  )
92
 
93
- pipe.set_adapters(["lightx2v", "lora_h", "lightx2v_2", "lora_l"], adapter_weights=[1., 1., 1., 1.])
94
- pipe.fuse_lora(adapter_names=["lightx2v"], lora_scale=0.9, components=["transformer"])
95
- pipe.fuse_lora(adapter_names=["lora_h"], lora_scale=0.4, components=["transformer"])
96
- pipe.fuse_lora(adapter_names=["lightx2v_2"], lora_scale=0.9, components=["transformer_2"])
97
- pipe.fuse_lora(adapter_names=["lora_l"], lora_scale=0.8, components=["transformer_2"])
98
  pipe.unload_lora_weights()
99
 
100
  quantize_(pipe.text_encoder, Int8WeightOnlyConfig())
 
61
  ).to('cuda')
62
 
63
  pipe.load_lora_weights(
64
+ "obsxrver/wan2.2-i2v-scat",
65
+ weight_name="WAN2.2-I2V-HighNoise_scat-xxi-i2v.safetensors",
66
+ adapter_name="i2v_scat"
67
  )
 
 
 
68
  pipe.load_lora_weights(
69
+ "lightx2v/Wan2.2-Lightning",
70
+ weight_name="Wan2.2-I2V-A14B-4steps-lora-rank64-Seko-V1/high_noise_model.safetensors",
71
+ adapter_name="lightx2v"
72
  )
73
 
 
 
74
  pipe.load_lora_weights(
75
+ "obsxrver/wan2.2-i2v-scat",
76
+ weight_name="WAN2.2-I2V-LowNoise_scat-xxi-i2v.safetensors",
77
+ adapter_name="i2v_scat_2",
78
+ load_into_transformer_2=True
79
  )
 
 
 
80
  pipe.load_lora_weights(
81
+ "lightx2v/Wan2.2-Lightning",
82
+ weight_name="Wan2.2-I2V-A14B-4steps-lora-rank64-Seko-V1/low_noise_model.safetensors",
83
+ adapter_name="lightx2v_2",
84
+ load_into_transformer_2=True
85
  )
86
 
87
+
88
+ pipe.set_adapters([ "i2v_scat","lightx2v","i2v_scat_2","lightx2v_2"], adapter_weights=[0.95, 0.9, 0.95, 0.9])
89
+ pipe.fuse_lora(adapter_names=["i2v_scat","lightx2v"], lora_scale=1., components=["transformer"])
90
+ pipe.fuse_lora(adapter_names=["i2v_scat_2","lightx2v_2"], lora_scale=1., components=["transformer_2"])
 
91
  pipe.unload_lora_weights()
92
 
93
  quantize_(pipe.text_encoder, Int8WeightOnlyConfig())