GiorgioV commited on
Commit
7af52fd
·
verified ·
1 Parent(s): 542231f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -21
app.py CHANGED
@@ -72,43 +72,27 @@ pipe.load_lora_weights(
72
 
73
  pipe.load_lora_weights(
74
  "GiorgioV/LoRA_for_WAN_22",
75
- weight_name="DR34ML4Y_I2V_14B_HIGH.safetensors",
76
  adapter_name="high_noise_lora",
77
  token=os.environ.get("HF_TOKEN")
78
  )
79
  # 新增:加载你提供的low noise LoRA
80
  pipe.load_lora_weights(
81
  "GiorgioV/LoRA_for_WAN_22",
82
- weight_name="DR34ML4Y_I2V_14B_LOW.safetensors",
83
  adapter_name="low_noise_lora",
84
  token=os.environ.get("HF_TOKEN"),
85
  load_into_transformer_2=True
86
  )
87
 
88
 
89
- pipe.load_lora_weights(
90
- "GiorgioV/LoRA_for_WAN_22",
91
- weight_name="wan-doubleblowjob-70epo-k3nk.safetensors",
92
- adapter_name="high_noise_lora1",
93
- token=os.environ.get("HF_TOKEN")
94
- )
95
- # 新增:加载你提供的low noise LoRA
96
- pipe.load_lora_weights(
97
- "GiorgioV/LoRA_for_WAN_22",
98
- weight_name="wan-doubleblowjob-70epo-k3nk.safetensors",
99
- adapter_name="low_noise_lora1",
100
- token=os.environ.get("HF_TOKEN"),
101
- load_into_transformer_2=True
102
- )
103
-
104
 
105
  pipe.set_adapters(["lightx2v", "lightx2v_2",
106
- "high_noise_lora", "low_noise_lora",
107
- "high_noise_lora1", "low_noise_lora1"], adapter_weights=[1.5, 1., 1., 1., 1., 1.])
108
  # 修改了lora_scale
109
- pipe.fuse_lora(adapter_names=["lightx2v", "high_noise_lora", "high_noise_lora1"], lora_scales=[3.0, 3.0, 3.0], components=["transformer"])
110
  # 修改了lora_scale
111
- pipe.fuse_lora(adapter_names=["lightx2v_2", "low_noise_lora", "low_noise_lora1"], lora_scales=[1.0, 1.0, 1.0], components=["transformer_2"])
112
 
113
 
114
 
 
72
 
73
  pipe.load_lora_weights(
74
  "GiorgioV/LoRA_for_WAN_22",
75
+ weight_name="wan22-k3nk4llinon3-16epoc-full-high-k3nk.safetensors",
76
  adapter_name="high_noise_lora",
77
  token=os.environ.get("HF_TOKEN")
78
  )
79
  # 新增:加载你提供的low noise LoRA
80
  pipe.load_lora_weights(
81
  "GiorgioV/LoRA_for_WAN_22",
82
+ weight_name="wan22-k3nk4llinon3-15epoc-full-low-k3nk.safetensors",
83
  adapter_name="low_noise_lora",
84
  token=os.environ.get("HF_TOKEN"),
85
  load_into_transformer_2=True
86
  )
87
 
88
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
89
 
90
  pipe.set_adapters(["lightx2v", "lightx2v_2",
91
+ "high_noise_lora", "low_noise_lora"], adapter_weights=[1., 1., 1., 1.])
 
92
  # 修改了lora_scale
93
+ pipe.fuse_lora(adapter_names=["lightx2v", "high_noise_lora"], lora_scales=[3.0, 3.0], components=["transformer"])
94
  # 修改了lora_scale
95
+ pipe.fuse_lora(adapter_names=["lightx2v_2", "low_noise_lora"], lora_scales=[1.0, 1.0], components=["transformer_2"])
96
 
97
 
98