From 226020b05a0ab6edd15aa24ec201a913e62348bf Mon Sep 17 00:00:00 2001
From: Luca Beltrame <lbeltrame@kde.org>
Date: Sat, 17 Jun 2023 11:39:46 +0200
Subject: [PATCH] Use a generic filename for the model

So it won't change if a newer version is uploaded.
---
 stable-diffusion/launch_training.sh | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/stable-diffusion/launch_training.sh b/stable-diffusion/launch_training.sh
index a13b4e3..4511e64 100644
--- a/stable-diffusion/launch_training.sh
+++ b/stable-diffusion/launch_training.sh
@@ -6,9 +6,11 @@ set -o pipefail
 
 output_name="$1"
 
+pushd /workspace/kohya_ss
+
 accelerate launch \
     --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" \
     --resolution=512 \
     --output_dir="/workspace/images/train/model" \
@@ -42,3 +44,5 @@ accelerate launch \
     --shuffle_caption \
     --xformers \
     --bucket_no_upscale
+
+popd