Zengtudor/Wan2.1
1
0
Fork 0
mirror of https://github.com/Wan-Video/Wan2.1.git synced 2025-12-16 12:13:27 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

added support for qwen lora safetensors format

Browse Source
This commit is contained in:
deepbeepmeep 2025-08-25 15:26:32 +02:00
parent 9b6448c19c
commit 7b3715f410
1 changed files with 29 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

38
models/qwen/transformer_qwenimage.py
View File

@ -497,16 +497,36 @@ class QwenImageTransformer2DModel(nn.Module):
new_sd[k] = v new_sd[k] = v
sd = new_sd sd = new_sd
if first.startswith("transformer_blocks"): prefix_list = ["lora_unet_transformer_blocks"]
new_sd = {} for prefix in prefix_list:
for k,v in sd.items(): if first.startswith(prefix):
if k.startswith("transformer_blocks"): repl_list = ["attn", "img_mlp", "txt_mlp", "img_mod", "txt_mod"]
k = "diffusion_model." + k src_list = ["_" + k + "_" for k in repl_list]
tgt_list = ["." + k + "." for k in repl_list]
src_list2 = ["_0_", "_0.", "_1.", "_2."]
tgt_list2 = [".0.", ".0.", ".1.", ".2."]
new_sd = {}
for k,v in sd.items():
k = "diffusion_model.transformer_blocks." + k[len(prefix)+1:]
for s,t in zip(src_list, tgt_list):
k = k.replace(s,t)
for s,t in zip(src_list2, tgt_list2):
k = k.replace(s,t)
new_sd[k] = v new_sd[k] = v
sd = new_sd sd = new_sd
return sd return sd
else:
return sd prefix_list = ["transformer_blocks"]
for prefix in prefix_list:
if first.startswith(prefix):
new_sd = {}
for k,v in sd.items():
if k.startswith(prefix):
k = "diffusion_model." + k
new_sd[k] = v
sd = new_sd
return sd
return sd
def __init__( def __init__(
self, self,