GiorgioV commited on
Commit
5dfb35e
·
verified ·
1 Parent(s): 3c0cd28

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -20
app.py CHANGED
@@ -85,29 +85,12 @@ pipe.load_lora_weights(
85
  load_into_transformer_2=True
86
  )
87
 
88
- pipe.load_lora_weights(
89
- "romanfratric234/WAN_2-1_Gangbang_Missionary_Sex_I2V",
90
- weight_name="gangbang-final-000002.safetensors",
91
- adapter_name="high_noise_lora_spec",
92
- token=os.environ.get("HF_TOKEN")
93
- )
94
- # 新增:加载你提供的low noise LoRA
95
- pipe.load_lora_weights(
96
- "romanfratric234/WAN_2-1_Gangbang_Missionary_Sex_I2V",
97
- weight_name="gangbang-final-000002.safetensors",
98
- adapter_name="low_noise_lora_spec",
99
- token=os.environ.get("HF_TOKEN"),
100
- load_into_transformer_2=True
101
- )
102
-
103
-
104
-
105
  pipe.set_adapters(["lightx2v", "lightx2v_2",
106
- "high_noise_lora", "low_noise_lora", "high_noise_lora_spec", "low_noise_lora_spec"], adapter_weights=[1.5, 1., 1., 1., 0.8, 1.])
107
  # 修改了lora_scale
108
- pipe.fuse_lora(adapter_names=["lightx2v", "high_noise_lora", "high_noise_lora_spec"], lora_scales=[3.0, 3.0, 3.0], components=["transformer"])
109
  # 修改了lora_scale
110
- pipe.fuse_lora(adapter_names=["lightx2v_2", "low_noise_lora", "low_noise_lora_spec"], lora_scales=[1.0, 1.0, 1.0], components=["transformer_2"])
111
 
112
 
113
 
 
85
  load_into_transformer_2=True
86
  )
87
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
88
  pipe.set_adapters(["lightx2v", "lightx2v_2",
89
+ "high_noise_lora", "low_noise_lora"], adapter_weights=[1., 1., 1., 1.])
90
  # 修改了lora_scale
91
+ pipe.fuse_lora(adapter_names=["lightx2v", "high_noise_lora"], lora_scales=[3.0, 3.0], components=["transformer"])
92
  # 修改了lora_scale
93
+ pipe.fuse_lora(adapter_names=["lightx2v_2", "low_noise_lora"], lora_scales=[1.0, 1.0], components=["transformer_2"])
94
 
95
 
96