From 55a34a371332ad2de440c05719652182efd0d950 Mon Sep 17 00:00:00 2001
From: Luca Beltrame <lbeltrame@kde.org>
Date: Sun, 4 Jun 2023 15:16:08 +0200
Subject: [PATCH] Script to download LoRA, LyCORIS and CKPTs to temporary
 folders

---
 stable-diffusion/download_temp_model.sh | 44 +++++++++++++++++++++++++
 1 file changed, 44 insertions(+)
 create mode 100644 stable-diffusion/download_temp_model.sh

diff --git a/stable-diffusion/download_temp_model.sh b/stable-diffusion/download_temp_model.sh
new file mode 100644
index 0000000..e8532ec
--- /dev/null
+++ b/stable-diffusion/download_temp_model.sh
@@ -0,0 +1,44 @@
+#!/bin/bash
+
+set -e
+set -o nounset
+set -o pipefail
+
+model_addr="$1"
+model_type="$2"
+sd_root="/notebooks/automatic/"
+
+case "${model_type}" in
+    "lora")
+        tmpdir="/tmp/lora"
+        dest="${sd_root}/models/Lora/"
+        ;;
+    "ckpt")
+        tmpdir="/tmp/models"
+        dest="${sd_root}/models/Stable-diffusion/"
+        ;;
+    "lycoris")
+        tmpdir="/tmp/models"
+        dest="${sd_root}/models/LyCORIS/"
+        ;;
+    *)
+        echo "Unknown model type."
+        exit 1
+        ;;
+esac
+
+mkdir -p "${tmpdir}"
+
+cd "${tmpdir}"
+
+modelfile="$(curl -O -J -L --remote-name -s \
+    -w "%{filename_effective}" "${model_addr}")"
+
+echo "Downloaded ${modelfile}"
+fullpath=$(realpath "${modelfile}")
+
+echo "Linking ${modelfile} to ${dest}..."
+
+pushd "$dest"
+ln -s "$fullpath" .
+popd