From 5413383f8ba3d8a4f7513b27cf2e3ddeb4ddd050 Mon Sep 17 00:00:00 2001
From: Ashley Kleynhans <ashley.kleynhans@gmail.com>
Date: Mon, 11 Mar 2024 12:59:04 +0200
Subject: [PATCH] Export HF_HOME before starting Kohya_ss

---
 Dockerfile                | 2 +-
 scripts/start_kohya_ss.sh | 1 +
 2 files changed, 2 insertions(+), 1 deletion(-)

diff --git a/Dockerfile b/Dockerfile
index 36a63a1..d09aea4 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -82,7 +82,7 @@ RUN python3 -m venv --system-site-packages venv && \
     source venv/bin/activate && \
     pip3 install torch==${TORCH_VERSION}+cu${CU_VERSION} torchvision torchaudio --index-url ${INDEX_URL} && \
     pip3 install xformers==${XFORMERS_VERSION}+cu${CU_VERSION} --index-url ${INDEX_URL} && \
-        bitsandbytes==0.41.2 \
+    pip3 install bitsandbytes==0.41.2 \
         tensorboard==2.15.2 \
         tensorflow==2.15.0.post1 \
         wheel \
diff --git a/scripts/start_kohya_ss.sh b/scripts/start_kohya_ss.sh
index b0f4352..7c3e91f 100755
--- a/scripts/start_kohya_ss.sh
+++ b/scripts/start_kohya_ss.sh
@@ -3,6 +3,7 @@
 echo "Starting Kohya_ss Web UI"
 cd /workspace/kohya_ss
 source venv/bin/activate
+export HF_HOME="/workspace"
 nohup ./gui.sh --listen 0.0.0.0 --server_port 3001 --headless > /workspace/logs/kohya_ss.log 2>&1 &
 echo "Kohya_ss started"
 echo "Log file: /workspace/logs/kohya_ss.log"