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