unsloth dockerfile

huggingface · Jul 24, 2024 · 72870ab · 72870ab
1 parent 88f91af
commit 72870ab
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/Dockerfile b/Dockerfile
@@ -80,5 +80,5 @@ RUN pip install -e . && \
     pip install -U ninja && \
     pip install -U flash-attn --no-build-isolation && \
     pip install -U deepspeed && \
-    pip install "unsloth[cu121-ampere-torch230] @ git+https://github.com/unslothai/unsloth.git" --no-deps && \
+    pip install --upgrade --force-reinstall --no-cache-dir "unsloth[cu121-ampere-torch230] @ git+https://github.com/unslothai/unsloth.git" --no-deps && \
     pip cache purge