diff --git a/examples/pytorch/stable_diffusion/custom/finetune_stable_diffusion_custom.py b/examples/pytorch/stable_diffusion/custom/finetune_stable_diffusion_custom.py index 35eb3792..47089bef 100644 --- a/examples/pytorch/stable_diffusion/custom/finetune_stable_diffusion_custom.py +++ b/examples/pytorch/stable_diffusion/custom/finetune_stable_diffusion_custom.py @@ -89,6 +89,13 @@ class StableDiffusionCustomArguments(TrainingArgs): 'help': 'The instance data local dir or online ID.', }) + concepts_list: str = field( + default=None, + metadata={ + 'help': 'Path to json containing multiple concepts.', + }) + + training_args = StableDiffusionCustomArguments( task='text-to-image-synthesis').parse_cli() @@ -134,6 +141,7 @@ kwargs = dict( sample_batch_size=args.sample_batch_size, initializer_token=args.initializer_token, class_data_dir=args.class_data_dir, + concepts_list=args.concepts_list, resolution=args.resolution, prior_loss_weight=args.prior_loss_weight, freeze_model=args.freeze_model, diff --git a/examples/pytorch/stable_diffusion/custom/run_train_custom.sh b/examples/pytorch/stable_diffusion/custom/run_train_custom.sh index fe8de203..3e1ff024 100644 --- a/examples/pytorch/stable_diffusion/custom/run_train_custom.sh +++ b/examples/pytorch/stable_diffusion/custom/run_train_custom.sh @@ -7,7 +7,8 @@ PYTHONPATH=. torchrun examples/pytorch/stable_diffusion/custom/finetune_stable_d --class_data_dir './tmp/class_data' \ --train_dataset_name 'buptwq/lora-stable-diffusion-finetune-dog' \ --max_epochs 250 \ - --modifier_token "" \ + --concepts_list '/mnt/user/E-yijing.wq-401594/github/custom_diffusion/modelscope/json/concept_list.json' \ + --modifier_token "+" \ --num_class_images=200 \ --save_ckpt_strategy 'by_epoch' \ --logging_interval 1 \