Compare commits

..

2 Commits

Author SHA1 Message Date
Artiprocher
fdeae36cfb xxx 2025-04-02 15:01:41 +08:00
Artiprocher
a2a720267e wan lora converter 2025-04-02 12:47:52 +08:00
2 changed files with 1 additions and 16 deletions

View File

@@ -365,22 +365,7 @@ class FluxLoRAConverter:
else:
state_dict_[name] = param
return state_dict_
class WanLoRAConverter:
def __init__(self):
pass
@staticmethod
def align_to_opensource_format(state_dict, **kwargs):
state_dict = {"diffusion_model." + name.replace(".default.", "."): param for name, param in state_dict.items()}
return state_dict
@staticmethod
def align_to_diffsynth_format(state_dict, **kwargs):
state_dict = {name.replace("diffusion_model.", "").replace(".lora_A.weight", ".lora_A.default.weight").replace(".lora_B.weight", ".lora_B.default.weight"): param for name, param in state_dict.items()}
return state_dict
def get_lora_loaders():
return [SDLoRAFromCivitai(), SDXLLoRAFromCivitai(), FluxLoRAFromCivitai(), HunyuanVideoLoRAFromCivitai(), GeneralLoRAFromPeft()]

View File

@@ -14,7 +14,7 @@ else:
setup(
name="diffsynth",
version="1.1.3",
version="1.1.2",
description="Enjoy the magic of Diffusion models!",
author="Artiprocher",
packages=find_packages(),