Mirrors/modelscope
1
0
Fork 0
mirror of https://github.com/modelscope/modelscope.git synced 2026-02-24 12:10:09 +01:00
Code Issues Packages Projects Releases Wiki Activity

fix bugs

Browse Source
This commit is contained in:
XDUWQ
2023-07-26 16:35:40 +08:00
parent 25d67a0b83
commit 99aa707995
1 changed files with 1 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
examples/pytorch/stable_diffusion/custom/run_train_custom.sh
View File

@@ -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 <new1> 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 "<new1>+<new2>" \
--num_class_images=200 \
--save_ckpt_strategy 'by_epoch' \