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"