Spaces:
Paused
Paused
Commit
·
08f4fb8
1
Parent(s):
e6b895f
mv url to correct position
Browse files- Dockerfile +2 -2
Dockerfile
CHANGED
@@ -34,8 +34,8 @@ ENV HOME=/home/user \
|
|
34 |
|
35 |
RUN pip3 install torch==2.3.1 torchvision==0.18.1 torchaudio==2.3.1 --index-url https://download.pytorch.org/whl/cu121
|
36 |
RUN pip3 install --no-cache-dir --upgrade -r /code/requirements.txt
|
37 |
-
RUN python /code/get_flash_attn.py > url.txt
|
38 |
-
RUN source
|
39 |
RUN pip3 install https://github.com/Dao-AILab/flash-attention/releases/download/v2.7.2.post1/${FLASH_ATTN_VER} --no-cache-dir
|
40 |
|
41 |
# Set the working directory to the user's home directory
|
|
|
34 |
|
35 |
RUN pip3 install torch==2.3.1 torchvision==0.18.1 torchaudio==2.3.1 --index-url https://download.pytorch.org/whl/cu121
|
36 |
RUN pip3 install --no-cache-dir --upgrade -r /code/requirements.txt
|
37 |
+
RUN python /code/get_flash_attn.py > /code/url.txt
|
38 |
+
RUN source /code/url.txt
|
39 |
RUN pip3 install https://github.com/Dao-AILab/flash-attention/releases/download/v2.7.2.post1/${FLASH_ATTN_VER} --no-cache-dir
|
40 |
|
41 |
# Set the working directory to the user's home directory
|