listen2you commited on
Commit
08f4fb8
·
1 Parent(s): e6b895f

mv url to correct position

Browse files
Files changed (1) hide show
  1. 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 ./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
 
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