Export HF_HOME before starting Kohya_ss

ashleykleynhans · Mar 11, 2024 · 5413383 · 5413383
1 parent e70396d
commit 5413383
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git 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
@@ -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"