GiorgioV commited on
Commit
a2e0009
·
verified ·
1 Parent(s): fbabdd0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +23 -4
app.py CHANGED
@@ -74,6 +74,14 @@ pipe.load_lora_weights(
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(
@@ -90,24 +98,35 @@ pipe.load_lora_weights(
90
  adapter_name="lora_l", **kwargs_lora_l
91
  )
92
 
 
 
 
 
 
 
 
 
93
  kwargs_lora_djl = {}
94
  kwargs_lora_djl["load_into_transformer_2"] = True
95
  pipe.load_lora_weights(
96
  "GiorgioV/LoRA_for_WAN_22",
97
  weight_name="DJ_I2V_14b_LowNoise.safetensors",
98
- adapter_name="lora_djl", **kwargs_lora_djl
99
  )
100
 
101
 
102
 
103
- pipe.set_adapters(["lightx2v", "lora_h", "lightx2v_2", "lora_l", "lora_djl"], adapter_weights=[1., 1., 1., 1., 1.])
104
  pipe.fuse_lora(adapter_names=["lightx2v"], lora_scale=3., components=["transformer"])
105
  pipe.fuse_lora(adapter_names=["lora_h"], lora_scale=0.3, components=["transformer"])
 
106
  pipe.fuse_lora(adapter_names=["lightx2v_2"], lora_scale=1., components=["transformer_2"])
107
- pipe.fuse_lora(adapter_names=["lora_l"], lora_scale=0.8, components=["transformer_2"])
108
- pipe.fuse_lora(adapter_names=["lora_djl"], lora_scale=0.8, components=["transformer_2"])
 
109
  pipe.unload_lora_weights()
110
 
 
111
  quantize_(pipe.text_encoder, Int8WeightOnlyConfig())
112
  quantize_(pipe.transformer, Float8DynamicActivationFloat8WeightConfig())
113
  quantize_(pipe.transformer_2, Float8DynamicActivationFloat8WeightConfig())
 
74
  adapter_name="lora_h", **kwargs_lora_h
75
  )
76
 
77
+ kwargs_lora_gh = {}
78
+ kwargs_lora_gh["load_into_transformer"] = True
79
+ pipe.load_lora_weights(
80
+ "GiorgioV/LoRA_for_WAN_22",
81
+ weight_name="Wan2.2 - I2V - GH - HIGH 14B.safetensors",
82
+ adapter_name="lora_gh", **kwargs_lora_gh
83
+ )
84
+
85
  kwargs_lora = {}
86
  kwargs_lora["load_into_transformer_2"] = True
87
  pipe.load_lora_weights(
 
98
  adapter_name="lora_l", **kwargs_lora_l
99
  )
100
 
101
+ kwargs_lora_gl = {}
102
+ kwargs_lora_gl["load_into_transformer_2"] = True
103
+ pipe.load_lora_weights(
104
+ "GiorgioV/LoRA_for_WAN_22",
105
+ weight_name="Wan2.2 - I2V - GH - LOW 14B.safetensors",
106
+ adapter_name="lora_gl", **kwargs_lora_gl
107
+ )
108
+
109
  kwargs_lora_djl = {}
110
  kwargs_lora_djl["load_into_transformer_2"] = True
111
  pipe.load_lora_weights(
112
  "GiorgioV/LoRA_for_WAN_22",
113
  weight_name="DJ_I2V_14b_LowNoise.safetensors",
114
+ adapter_name="lora_gjl", **kwargs_lora_djl
115
  )
116
 
117
 
118
 
119
+ pipe.set_adapters(["lightx2v", "lora_h", "lora_gh", "lightx2v_2", "lora_l", "lora_gl", "lora_gjl"], adapter_weights=[1., 1., 1., 1., 1., 1., 1.])
120
  pipe.fuse_lora(adapter_names=["lightx2v"], lora_scale=3., components=["transformer"])
121
  pipe.fuse_lora(adapter_names=["lora_h"], lora_scale=0.3, components=["transformer"])
122
+ pipe.fuse_lora(adapter_names=["lora_gh"], lora_scale=0.3, components=["transformer"])
123
  pipe.fuse_lora(adapter_names=["lightx2v_2"], lora_scale=1., components=["transformer_2"])
124
+ pipe.fuse_lora(adapter_names=["lora_l"], lora_scale=1., components=["transformer_2"])
125
+ pipe.fuse_lora(adapter_names=["lora_gl"], lora_scale=0.8, components=["transformer_2"])
126
+ pipe.fuse_lora(adapter_names=["lora_gjl"], lora_scale=0.8, components=["transformer_2"])
127
  pipe.unload_lora_weights()
128
 
129
+
130
  quantize_(pipe.text_encoder, Int8WeightOnlyConfig())
131
  quantize_(pipe.transformer, Float8DynamicActivationFloat8WeightConfig())
132
  quantize_(pipe.transformer_2, Float8DynamicActivationFloat8WeightConfig())