From ffbf77fcf25c4253dd850ae1d83fd54cbec61c28 Mon Sep 17 00:00:00 2001
From: XDUWQ <1300964705@qq.com>
Date: Tue, 25 Jul 2023 14:47:45 +0800
Subject: [PATCH] update

---
 .../custom/finetune_stable_diffusion_custom.py            | 8 ++++++++
 .../pytorch/stable_diffusion/custom/run_train_custom.sh   | 3 ++-
 2 files changed, 10 insertions(+), 1 deletion(-)

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 "<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 \