GiorgioV commited on
Commit
f7a354b
·
verified ·
1 Parent(s): 6e3d0d9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -18
app.py CHANGED
@@ -70,14 +70,6 @@ pipe.load_lora_weights(
70
  adapter_name="lightx2v"
71
  )
72
 
73
- kwargs_lora_hh = {}
74
- kwargs_lora_hh["load_into_transformer"] = True
75
- pipe.load_lora_weights(
76
- "GiorgioV/LoRA_for_WAN_22",
77
- weight_name="P003-SD-i2v-v10-000010_converted.safetensors",
78
- adapter_name="lora_hh", **kwargs_lora_hh
79
- )
80
-
81
  kwargs_lora = {}
82
  kwargs_lora["load_into_transformer_2"] = True
83
  pipe.load_lora_weights(
@@ -86,19 +78,10 @@ pipe.load_lora_weights(
86
  adapter_name="lightx2v_2", **kwargs_lora
87
  )
88
 
89
- kwargs_lora_lh = {}
90
- kwargs_lora_lh["load_into_transformer_2"] = True
91
- pipe.load_lora_weights(
92
- "GiorgioV/LoRA_for_WAN_22",
93
- weight_name="P003-SD-i2v-v10-000010_converted.safetensors",
94
- adapter_name="lora_lh", **kwargs_lora_lh
95
- )
96
 
97
- pipe.set_adapters(["lightx2v", "lora_hh", "lightx2v_2", "lora_lh"], adapter_weights=[1., 1., 1., 1.])
98
  pipe.fuse_lora(adapter_names=["lightx2v"], lora_scale=3., components=["transformer"])
99
- pipe.fuse_lora(adapter_names=["lora_hh"], lora_scale=0.3, components=["transformer"])
100
  pipe.fuse_lora(adapter_names=["lightx2v_2"], lora_scale=1., components=["transformer_2"])
101
- pipe.fuse_lora(adapter_names=["lora_lh"], lora_scale=0.3, components=["transformer_2"])
102
  pipe.unload_lora_weights()
103
 
104
  quantize_(pipe.text_encoder, Int8WeightOnlyConfig())
 
70
  adapter_name="lightx2v"
71
  )
72
 
 
 
 
 
 
 
 
 
73
  kwargs_lora = {}
74
  kwargs_lora["load_into_transformer_2"] = True
75
  pipe.load_lora_weights(
 
78
  adapter_name="lightx2v_2", **kwargs_lora
79
  )
80
 
 
 
 
 
 
 
 
81
 
82
+ pipe.set_adapters(["lightx2v", "lightx2v_2"], adapter_weights=[1., 1.])
83
  pipe.fuse_lora(adapter_names=["lightx2v"], lora_scale=3., components=["transformer"])
 
84
  pipe.fuse_lora(adapter_names=["lightx2v_2"], lora_scale=1., components=["transformer_2"])
 
85
  pipe.unload_lora_weights()
86
 
87
  quantize_(pipe.text_encoder, Int8WeightOnlyConfig())