theluyuan/DiffSynth-Studio
1
0
Fork 0
mirror of https://github.com/modelscope/DiffSynth-Studio.git synced 2026-03-19 06:23:43 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #210 from modelscope/opensource-alignment

Browse Source 
staticmethod
This commit is contained in:
Zhongjie Duan
2024-09-14 17:18:19 +08:00
committed by GitHub
parent 884dd749a0 3da8aa257b
commit 44a8bf4143
2 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
diffsynth/models/lora.py
View File

@@ -252,7 +252,8 @@ class FluxLoRAConverter:
def __init__(self):
pass
def align_to_opensource_format(self, state_dict, alpha=1.0):
@staticmethod
def align_to_opensource_format(state_dict, alpha=1.0):
prefix_rename_dict = {
"single_blocks": "lora_unet_single_blocks",
"blocks": "lora_unet_double_blocks",

2
examples/train/flux/train_flux_lora.py
View File

@@ -81,6 +81,6 @@ if __name__ == '__main__':
lora_rank=args.lora_rank,
lora_alpha=args.lora_alpha,
lora_target_modules=args.lora_target_modules,
state_dict_converter=FluxLoRAConverter().align_to_opensource_format if args.align_to_opensource_format else None,
state_dict_converter=FluxLoRAConverter.align_to_opensource_format if args.align_to_opensource_format else None,
)
launch_training_task(model, args)