mirror of
https://github.com/modelscope/modelscope.git
synced 2025-12-22 19:19:21 +01:00
13 lines
552 B
Bash
13 lines
552 B
Bash
PYTHONPATH=. torchrun examples/pytorch/stable_diffusion/finetune_stable_diffusion.py \
|
|
--model 'damo/multi-modal_efficient-diffusion-tuning-lora' \
|
|
--work_dir './tmp/stable_diffusion_tuning' \
|
|
--train_dataset_namespace 'damo' \
|
|
--train_dataset_name 'buptwq/lora-stable-diffusion-finetune-dog' \
|
|
--max_epochs 150 \
|
|
--save_ckpt_strategy 'by_epoch' \
|
|
--logging_interval 100 \
|
|
--train.dataloader.workers_per_gpu 0 \
|
|
--evaluation.dataloader.workers_per_gpu 0 \
|
|
--train.optimizer.lr 1e-4 \
|
|
--use_model_config true
|