diff --git a/train.bat b/train.bat deleted file mode 100644 index b52c776..0000000 --- a/train.bat +++ /dev/null @@ -1 +0,0 @@ -python main.py --base "configs/stable-diffusion/v1-finetune.yaml" -t --no-test --actual-resume "./models/ldm/stable-diffusion/model.ckpt" -n "waifu-diffusion" --gpus 1 --data_root "./danbooru-aesthetic" \ No newline at end of file diff --git a/train.sh b/train.sh new file mode 100644 index 0000000..b35545b --- /dev/null +++ b/train.sh @@ -0,0 +1 @@ +python3 main.py --resume model.ckpt --base ./configs/stable-diffusion/v1-finetune-4gpu.yaml --no-test --seed 25 --scale_lr False --gpus 0,1,2,3