update

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,

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 "<new1>" \
+    --concepts_list '/mnt/user/E-yijing.wq-401594/github/custom_diffusion/modelscope/json/concept_list.json' \
+    --modifier_token "<new1>+<new2>" \
     --num_class_images=200 \
     --save_ckpt_strategy 'by_epoch' \
     --logging_interval 1 \