mirror of
https://github.com/modelscope/modelscope.git
synced 2026-02-25 04:30:48 +01:00
0db0ec5586
1. Merge(add) daily regression from github PR (daily_regression.yaml) 2. Add lora stable diffusion from github PR Link: https://code.alibaba-inc.com/Ali-MaaS/MaaS-lib/codereview/13010802 * fix: device arg not work, rename device to ngpu (#272) * Correcting the lora stable diffusion example script (#300) * add vad model and punc model in README.md add vad model and punc model * Merge pull request #302 from modelscope/langgz-patch-1 add vad model and punc model in README.md * add 1.6 * modify ignore * Merge pull request #307 from modelscope/dev_rs_16 Merge release 1.6 * undo datetime to 2099 * Merge pull request #311 from modelscope/fix_master_version undo datetime to 2099 * add daily regression workflow * modify workflow name * fix cron format issue * lora trainer * Merge pull request #315 from liuyhwangyh/add_regression_workflow add daily regression workflow
13 lines
519 B
Bash
13 lines
519 B
Bash
PYTHONPATH=. torchrun examples/pytorch/stable_diffusion/finetune_stable_diffusion.py \
|
|
--model 'AI-ModelScope/stable-diffusion-v1-5' \
|
|
--model_revision 'v1.0.6' \
|
|
--work_dir './tmp/lora_diffusion' \
|
|
--train_dataset_name 'buptwq/lora-stable-diffusion-finetune' \
|
|
--max_epochs 100 \
|
|
--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
|