From 42ec7b08eb1a895c5c0a3528cca8b2b5504d158e Mon Sep 17 00:00:00 2001 From: Artiprocher Date: Thu, 4 Sep 2025 15:45:39 +0800 Subject: [PATCH] bugfix --- examples/flux/model_training/train.py | 2 -- examples/wanvideo/model_training/train.py | 2 -- 2 files changed, 4 deletions(-) diff --git a/examples/flux/model_training/train.py b/examples/flux/model_training/train.py index 568c77a..4a82228 100644 --- a/examples/flux/model_training/train.py +++ b/examples/flux/model_training/train.py @@ -117,6 +117,4 @@ if __name__ == "__main__": 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, ) - 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) diff --git a/examples/wanvideo/model_training/train.py b/examples/wanvideo/model_training/train.py index b9b7d8c..37494e7 100644 --- a/examples/wanvideo/model_training/train.py +++ b/examples/wanvideo/model_training/train.py @@ -126,6 +126,4 @@ if __name__ == "__main__": args.output_path, 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)