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

bugfix

Browse Source
This commit is contained in:
Artiprocher
2025-09-04 15:45:39 +08:00
parent d049fb6d1d
commit 42ec7b08eb
2 changed files with 0 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -117,6 +117,4 @@ if __name__ == "__main__":
remove_prefix_in_ckpt=args.remove_prefix_in_ckpt, remove_prefix_in_ckpt=args.remove_prefix_in_ckpt,
state_dict_converter=FluxLoRAConverter.align_to_opensource_format if args.align_to_opensource_format else lambda x:x, state_dict_converter=FluxLoRAConverter.align_to_opensource_format if args.align_to_opensource_format else lambda x:x,
) )
optimizer = torch.optim.AdamW(model.trainable_modules(), lr=args.learning_rate, weight_decay=args.weight_decay)
scheduler = torch.optim.lr_scheduler.ConstantLR(optimizer)
launch_training_task(dataset, model, model_logger, args=args) launch_training_task(dataset, model, model_logger, args=args)

2
examples/wanvideo/model_training/train.py
View File

@@ -126,6 +126,4 @@ if __name__ == "__main__":
args.output_path, args.output_path,
remove_prefix_in_ckpt=args.remove_prefix_in_ckpt remove_prefix_in_ckpt=args.remove_prefix_in_ckpt
) )
optimizer = torch.optim.AdamW(model.trainable_modules(), lr=args.learning_rate, weight_decay=args.weight_decay)
scheduler = torch.optim.lr_scheduler.ConstantLR(optimizer)
launch_training_task(dataset, model, model_logger, args=args) launch_training_task(dataset, model, model_logger, args=args)