mirror of
https://github.com/modelscope/DiffSynth-Studio.git
synced 2026-03-20 15:48:20 +00:00
Compare commits
2 Commits
v1.1.3
...
wan-lora-f
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
fdeae36cfb | ||
|
|
a2a720267e |
@@ -365,22 +365,7 @@ class FluxLoRAConverter:
|
|||||||
else:
|
else:
|
||||||
state_dict_[name] = param
|
state_dict_[name] = param
|
||||||
return state_dict_
|
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():
|
def get_lora_loaders():
|
||||||
return [SDLoRAFromCivitai(), SDXLLoRAFromCivitai(), FluxLoRAFromCivitai(), HunyuanVideoLoRAFromCivitai(), GeneralLoRAFromPeft()]
|
return [SDLoRAFromCivitai(), SDXLLoRAFromCivitai(), FluxLoRAFromCivitai(), HunyuanVideoLoRAFromCivitai(), GeneralLoRAFromPeft()]
|
||||||
|
|||||||
2
setup.py
2
setup.py
@@ -14,7 +14,7 @@ else:
|
|||||||
|
|
||||||
setup(
|
setup(
|
||||||
name="diffsynth",
|
name="diffsynth",
|
||||||
version="1.1.3",
|
version="1.1.2",
|
||||||
description="Enjoy the magic of Diffusion models!",
|
description="Enjoy the magic of Diffusion models!",
|
||||||
author="Artiprocher",
|
author="Artiprocher",
|
||||||
packages=find_packages(),
|
packages=find_packages(),
|
||||||
|
|||||||
Reference in New Issue
Block a user