GiorgioV commited on
Commit
069d945
·
verified ·
1 Parent(s): b4f0e9d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +19 -1
app.py CHANGED
@@ -70,6 +70,14 @@ pipe.load_lora_weights(
70
  adapter_name="lightx2v"
71
  )
72
 
 
 
 
 
 
 
 
 
73
  kwargs_lora = {}
74
  kwargs_lora["load_into_transformer_2"] = True
75
  pipe.load_lora_weights(
@@ -78,9 +86,19 @@ pipe.load_lora_weights(
78
  adapter_name="lightx2v_2", **kwargs_lora
79
  )
80
 
81
- pipe.set_adapters(["lightx2v", "lightx2v_2"], adapter_weights=[1., 1.])
 
 
 
 
 
 
 
 
82
  pipe.fuse_lora(adapter_names=["lightx2v"], lora_scale=3., components=["transformer"])
 
83
  pipe.fuse_lora(adapter_names=["lightx2v_2"], lora_scale=1., components=["transformer_2"])
 
84
  pipe.unload_lora_weights()
85
 
86
  quantize_(pipe.text_encoder, Int8WeightOnlyConfig())
 
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
  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=1., 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=1., components=["transformer_2"])
102
  pipe.unload_lora_weights()
103
 
104
  quantize_(pipe.text_encoder, Int8WeightOnlyConfig())