GiorgioV commited on
Commit
0a151ae
·
verified ·
1 Parent(s): ebcb9a7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +11 -1
app.py CHANGED
@@ -66,6 +66,15 @@ pipe.load_lora_weights(
66
  weight_name="Lightx2v/lightx2v_I2V_14B_480p_cfg_step_distill_rank128_bf16.safetensors",
67
  adapter_name="lightx2v"
68
  )
 
 
 
 
 
 
 
 
 
69
  kwargs_lora = {}
70
  kwargs_lora["load_into_transformer_2"] = True
71
  pipe.load_lora_weights(
@@ -82,8 +91,9 @@ pipe.load_lora_weights(
82
  adapter_name="lora_exp", **kwargs_lora_exp
83
  )
84
 
85
- pipe.set_adapters(["lightx2v", "lightx2v_2", "lora_exp"], adapter_weights=[1., 1., 1.])
86
  pipe.fuse_lora(adapter_names=["lightx2v"], lora_scale=3., components=["transformer"])
 
87
  pipe.fuse_lora(adapter_names=["lightx2v_2"], lora_scale=1., components=["transformer_2"])
88
  pipe.fuse_lora(adapter_names=["lora_exp"], lora_scale=1., components=["transformer_2"])
89
  pipe.unload_lora_weights()
 
66
  weight_name="Lightx2v/lightx2v_I2V_14B_480p_cfg_step_distill_rank128_bf16.safetensors",
67
  adapter_name="lightx2v"
68
  )
69
+
70
+ kwargs_lora_exp_h = {}
71
+ kwargs_lora_exp_h["load_into_transformer"] = True
72
+ pipe.load_lora_weights(
73
+ "GiorgioV/LoRA_for_WAN_22",
74
+ weight_name="LoRA_exp.safetensors",
75
+ adapter_name="lora_exp_h", **kwargs_lora_exp_h
76
+ )
77
+
78
  kwargs_lora = {}
79
  kwargs_lora["load_into_transformer_2"] = True
80
  pipe.load_lora_weights(
 
91
  adapter_name="lora_exp", **kwargs_lora_exp
92
  )
93
 
94
+ pipe.set_adapters(["lightx2v", "lora_exp_h", "lightx2v_2", "lora_exp"], adapter_weights=[1., 1., 1., 1.])
95
  pipe.fuse_lora(adapter_names=["lightx2v"], lora_scale=3., components=["transformer"])
96
+ pipe.fuse_lora(adapter_names=["lora_exp_h"], lora_scale=1., components=["transformer"])
97
  pipe.fuse_lora(adapter_names=["lightx2v_2"], lora_scale=1., components=["transformer_2"])
98
  pipe.fuse_lora(adapter_names=["lora_exp"], lora_scale=1., components=["transformer_2"])
99
  pipe.unload_lora_weights()