Upgrade stable diffusion version to a more powerful version 2.1 (#415)

This commit is contained in:
Wang Qiang
2023-07-27 20:40:38 +08:00
committed by GitHub
parent dca6143b8b
commit 34ea2b474a
2 changed files with 2 additions and 2 deletions

View File

@@ -1,5 +1,5 @@
PYTHONPATH=. torchrun examples/pytorch/stable_diffusion/dreambooth/finetune_stable_diffusion_dreambooth.py \
--model 'AI-ModelScope/stable-diffusion-v1-5' \
--model 'AI-ModelScope/stable-diffusion-v2-1' \
--model_revision 'v1.0.8' \
--work_dir './tmp/dreambooth_diffusion' \
--train_dataset_name 'buptwq/lora-stable-diffusion-finetune' \

View File

@@ -1,5 +1,5 @@
PYTHONPATH=. torchrun examples/pytorch/stable_diffusion/lora/finetune_stable_diffusion_lora.py \
--model 'AI-ModelScope/stable-diffusion-v1-5' \
--model 'AI-ModelScope/stable-diffusion-v2-1' \
--model_revision 'v1.0.9' \
--prompt "a dog" \
--work_dir './tmp/lora_diffusion' \