Hugging Face's logo

Spaces:
becteur92
/
smollvm
Paused

App Files Community
youssef commited on
Commit
43dbecd
·
1 Parent(s): 7516878

prebuilt

Browse files
Files changed (1) hide show
  1. Dockerfile +1 -1
Dockerfile CHANGED
@@ -43,7 +43,7 @@ RUN pyenv install ${PYTHON_VERSION} && \
43
 
44
  COPY --chown=1000 ./requirements.txt /tmp/requirements.txt
45
  RUN pip install --no-cache-dir --upgrade -r /tmp/requirements.txt && \
46
- pip install --no-cache-dir https://github.com/mjun0812/flash-attention-prebuild-wheels/releases/download/v0.0.5/flash_attn-2.7.4.post1+cu124torch2.1-cp310-cp310-linux_x86_64.whl
47
 
48
  COPY --chown=1000 . ${HOME}/app
49
  ENV PYTHONPATH=${HOME}/app \
 
43
 
44
  COPY --chown=1000 ./requirements.txt /tmp/requirements.txt
45
  RUN pip install --no-cache-dir --upgrade -r /tmp/requirements.txt && \
46
+ pip install --no-cache-dir https://github.com/mjun0812/flash-attention-prebuild-wheels/releases/download/v0.0.5/flash_attn-2.6.3+cu124torch2.1-cp310-cp310-linux_x86_64.whl
47
 
48
  COPY --chown=1000 . ${HOME}/app
49
  ENV PYTHONPATH=${HOME}/app \