fix

huggingface · Jun 13, 2024 · cbd1b95 · cbd1b95
1 parent f1f9040
commit cbd1b95
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/Dockerfile b/Dockerfile
@@ -77,7 +77,7 @@ COPY --chown=1000:1000 . /app/
 
 RUN pip install -e . && \
     python -m nltk.downloader punkt && \
-    pip install -U flash-attn --no-build-isolation && \
+    FLASH_ATTENTION_SKIP_CUDA_BUILD=TRUE 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 cache purge