theluyuan/DiffSynth-Studio
1
0
Fork 0
mirror of https://github.com/modelscope/DiffSynth-Studio.git synced 2026-03-24 01:48:13 +00:00
Code Issues Packages Projects Releases Wiki Activity

staticmethod

Browse Source
This commit is contained in:
Artiprocher
2024-09-14 17:16:59 +08:00
parent 884dd749a0
commit 3da8aa257b
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): def __init__(self):
pass 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 = { prefix_rename_dict = {
"single_blocks": "lora_unet_single_blocks", "single_blocks": "lora_unet_single_blocks",
"blocks": "lora_unet_double_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_rank=args.lora_rank,
lora_alpha=args.lora_alpha, lora_alpha=args.lora_alpha,
lora_target_modules=args.lora_target_modules, 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) launch_training_task(model, args)