einar/scripts
1
0
Fork 0
Code Releases Activity

Use a generic filename for the model

Browse source 
So it won't change if a newer version is uploaded.
This commit is contained in:
Luca Beltrame 2023-06-17 11:39:46 +02:00
parent 40183894f8
commit 226020b05a
Signed by: einar
GPG key ID: 4707F46E9EC72DEC
1 changed files with 5 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
stable-diffusion/launch_training.sh
View file

@ -6,9 +6,11 @@ set -o pipefail
output_name="$1" output_name="$1"
pushd /workspace/kohya_ss
accelerate launch \ accelerate launch \
--num_cpu_threads_per_process=10 "train_network.py" \ --num_cpu_threads_per_process=10 "train_network.py" \
--enable_bucket --pretrained_model_name_or_path="/workspace/AnyLoRA_noVae_fp16-pruned.ckpt" \ --enable_bucket --pretrained_model_name_or_path="/workspace/model.safetensors" \
--train_data_dir="/workspace/images/train/img" \ --train_data_dir="/workspace/images/train/img" \
--resolution=512 \ --resolution=512 \
--output_dir="/workspace/images/train/model" \ --output_dir="/workspace/images/train/model" \
@ -42,3 +44,5 @@ accelerate launch \
--shuffle_caption \ --shuffle_caption \
--xformers \ --xformers \
--bucket_no_upscale --bucket_no_upscale
popd