theluyuan/DiffSynth-Studio
1
0
Fork 0
mirror of https://github.com/modelscope/DiffSynth-Studio.git synced 2026-03-22 16:50:47 +00:00
Code Issues Packages Projects Releases Wiki Activity

support resume from opensource format

Browse Source
This commit is contained in:
Artiprocher
2024-12-16 12:25:05 +08:00
parent 8c2671ce40
commit 919d399fdb
3 changed files with 52 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -40,7 +40,8 @@ class LightningModel(LightningModelForT2ILoRA):
lora_alpha=lora_alpha,
lora_target_modules=lora_target_modules,
init_lora_weights=init_lora_weights,
pretrained_lora_path=pretrained_lora_path
pretrained_lora_path=pretrained_lora_path,
state_dict_converter=FluxLoRAConverter.align_to_diffsynth_format
)