NeoHaggis commited on
Commit
2ce3456
·
verified ·
1 Parent(s): 702b1e3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -37
app.py CHANGED
@@ -75,58 +75,27 @@ pipe.load_lora_weights(
75
 
76
  pipe.load_lora_weights(
77
  "GiorgioV/LoRA_for_WAN_22",
78
- weight_name="DR34ML4Y_I2V_14B_HIGH.safetensors",
79
  adapter_name="high_noise_lora",
80
  token=os.environ.get("HF_TOKEN")
81
  )
82
  # 新增:加载你提供的low noise LoRA
83
  pipe.load_lora_weights(
84
- "GiorgioV/LoRA_for_WAN_22",
85
- weight_name="DR34ML4Y_I2V_14B_LOW.safetensors",
86
  adapter_name="low_noise_lora",
87
  token=os.environ.get("HF_TOKEN"),
88
  load_into_transformer_2=True
89
  )
90
 
91
 
92
- pipe.load_lora_weights(
93
- "KGhaleon/POV_Miss",
94
- weight_name="wan2.2_i2v_highnoise_pov_missionary_v1.0.safetensors",
95
- adapter_name="high_noise_lora1",
96
- token=os.environ.get("HF_TOKEN")
97
- )
98
- # 新增:加载你提供的low noise LoRA
99
- pipe.load_lora_weights(
100
- "KGhaleon/POV_Miss",
101
- weight_name="wan2.2_i2v_lownoise_pov_missionary_v1.0.safetensors",
102
- adapter_name="low_noise_lora1",
103
- token=os.environ.get("HF_TOKEN"),
104
- load_into_transformer_2=True
105
- )
106
-
107
- pipe.load_lora_weights(
108
- "KeyOpening8063587/ZoomRevealI2V",
109
- weight_name="wan22-zoom-reveal-400epoc-high-k3nk.safetensors",
110
- adapter_name="high_noise_lora2",
111
- token=os.environ.get("HF_TOKEN")
112
- )
113
- # 新增:加载你提供的low noise LoRA
114
- pipe.load_lora_weights(
115
- "KeyOpening8063587/ZoomRevealI2V",
116
- weight_name="wan22-zoom-reveal-295epoc-low-k3nk.safetensors",
117
- adapter_name="low_noise_lora2",
118
- token=os.environ.get("HF_TOKEN"),
119
- load_into_transformer_2=True
120
- )
121
 
122
  pipe.set_adapters(["lightx2v", "lightx2v_2",
123
- "high_noise_lora", "low_noise_lora",
124
- "high_noise_lora1", "low_noise_lora1",
125
- "high_noise_lora2", "low_noise_lora2"], adapter_weights=[1.5, 1., 1., 1., 1., 1., 0.6, 0.6])
126
  # 修改了lora_scale
127
- pipe.fuse_lora(adapter_names=["lightx2v", "high_noise_lora", "high_noise_lora1", "high_noise_lora2"], lora_scales=[3.0, 3.0, 3.0, 3.0], components=["transformer"])
128
  # 修改了lora_scale
129
- pipe.fuse_lora(adapter_names=["lightx2v_2", "low_noise_lora", "low_noise_lora1", "low_noise_lora2"], lora_scales=[1.0, 1.0, 1.0, 1.0], components=["transformer_2"])
130
 
131
 
132
 
 
75
 
76
  pipe.load_lora_weights(
77
  "GiorgioV/LoRA_for_WAN_22",
78
+ weight_name="DR34ML4Y_I2V_14B_HIGH_V2.safetensors",
79
  adapter_name="high_noise_lora",
80
  token=os.environ.get("HF_TOKEN")
81
  )
82
  # 新增:加载你提供的low noise LoRA
83
  pipe.load_lora_weights(
84
+ "Passman22231/DR34ML4Y_I2V_14B_LOW_V2",
85
+ weight_name="DR34ML4Y_I2V_14B_LOW_V2.safetensors",
86
  adapter_name="low_noise_lora",
87
  token=os.environ.get("HF_TOKEN"),
88
  load_into_transformer_2=True
89
  )
90
 
91
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
92
 
93
  pipe.set_adapters(["lightx2v", "lightx2v_2",
94
+ "high_noise_lora", "low_noise_lora"], adapter_weights=[1., 1., 1., 0.8])
 
 
95
  # 修改了lora_scale
96
+ pipe.fuse_lora(adapter_names=["lightx2v", "high_noise_lora"], lora_scales=[3.0, 3.0], components=["transformer"])
97
  # 修改了lora_scale
98
+ pipe.fuse_lora(adapter_names=["lightx2v_2", "low_noise_lora"], lora_scales=[1.0, 1.0], components=["transformer_2"])
99
 
100
 
101