Mirrors/modelscope
1
0
Fork 0
mirror of https://github.com/modelscope/modelscope.git synced 2026-05-18 05:05:00 +02:00
Code Issues Packages Projects Releases Wiki Activity

custom diffusion

Browse Source
This commit is contained in:
XDUWQ
2023-07-12 15:02:43 +08:00
parent 0dc57f8dec
commit d6368b2617
3 changed files with 17 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
examples/pytorch/stable_diffusion/custom/finetune_stable_diffusion_custom.py
View File

@@ -156,7 +156,7 @@ pipe = pipeline(
modifier_token='<new1>',
model_revision=args.model_revision)
output = pipe({'text': args.prompt})
output = pipe({'text': args.instance_prompt})
# visualize the result on ipynb and save it
output
cv2.imwrite('./custom_result.png', output['output_imgs'][0])

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

@@ -6,7 +6,7 @@ PYTHONPATH=. torchrun examples/pytorch/stable_diffusion/custom/finetune_stable_d
--work_dir './tmp/custom_diffusion' \
--class_data_dir './tmp/class_data' \
--train_dataset_name 'buptwq/lora-stable-diffusion-finetune-dog' \
--max_epochs 1 \
--max_epochs 250 \
--modifier_token "<new1>" \
--num_class_images=200 \
--save_ckpt_strategy 'by_epoch' \