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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +10 -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(
@@ -79,8 +87,9 @@ pipe.load_lora_weights(
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
 
 
70
  adapter_name="lightx2v"
71
  )
72
 
73
+ kwargs_lora_h = {}
74
+ kwargs_lora_h["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_h", **kwargs_lora_h
79
+ )
80
+
81
  kwargs_lora = {}
82
  kwargs_lora["load_into_transformer_2"] = True
83
  pipe.load_lora_weights(
 
87
  )
88
 
89
 
90
+ pipe.set_adapters(["lightx2v", "lora_h", "lightx2v_2"], adapter_weights=[1., 1., 1.])
91
  pipe.fuse_lora(adapter_names=["lightx2v"], lora_scale=3., components=["transformer"])
92
+ pipe.fuse_lora(adapter_names=["lora_h"], lora_scale=0.3, components=["transformer"])
93
  pipe.fuse_lora(adapter_names=["lightx2v_2"], lora_scale=1., components=["transformer_2"])
94
  pipe.unload_lora_weights()
95