From 99aa7079952afa677b033f35fdb5df2c666826ee Mon Sep 17 00:00:00 2001 From: XDUWQ <1300964705@qq.com> Date: Wed, 26 Jul 2023 16:35:40 +0800 Subject: [PATCH] fix bugs --- examples/pytorch/stable_diffusion/custom/run_train_custom.sh | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/examples/pytorch/stable_diffusion/custom/run_train_custom.sh b/examples/pytorch/stable_diffusion/custom/run_train_custom.sh index 3e1ff024..fab8e059 100644 --- a/examples/pytorch/stable_diffusion/custom/run_train_custom.sh +++ b/examples/pytorch/stable_diffusion/custom/run_train_custom.sh @@ -1,5 +1,5 @@ PYTHONPATH=. torchrun examples/pytorch/stable_diffusion/custom/finetune_stable_diffusion_custom.py \ - --model 'AI-ModelScope/stable-diffusion-v1-5' \ + --model 'AI-ModelScope/stable-diffusion-v2-1' \ --model_revision 'v1.0.9' \ --class_prompt "dog" \ --instance_prompt="photo of a dog" \ @@ -7,7 +7,6 @@ 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 \ - --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' \