Skip to content

Commit

Permalink
Trigger build (#293)
Browse files Browse the repository at this point in the history
  • Loading branch information
vwxyzjn authored Aug 26, 2024
1 parent 8445178 commit e77830e
Show file tree
Hide file tree
Showing 2 changed files with 2 additions and 2 deletions.
2 changes: 1 addition & 1 deletion Dockerfile
Original file line number Diff line number Diff line change
Expand Up @@ -83,8 +83,8 @@ ENTRYPOINT ["bash", "-l"]
WORKDIR /stage/

# TODO When updating flash-attn or torch in the future, make sure to update the version in the requirements.txt file.
ENV HF_HUB_ENABLE_HF_TRANSFER=1
COPY requirements.txt .
ENV HF_HUB_ENABLE_HF_TRANSFER=1
RUN pip install --upgrade pip "setuptools<70.0.0" wheel
# TODO, unpin setuptools when this issue in flash attention is resolved
RUN pip install torch==2.3.0 torchvision==0.18.0 torchaudio==2.3.0 --index-url https://download.pytorch.org/whl/cu121
Expand Down
2 changes: 1 addition & 1 deletion Dockerfile.olmo
Original file line number Diff line number Diff line change
Expand Up @@ -83,8 +83,8 @@ ENTRYPOINT ["bash", "-l"]
WORKDIR /stage/

# TODO When updating flash-attn or torch in the future, make sure to update the version in the requirements.txt file.
ENV HF_HUB_ENABLE_HF_TRANSFER=1
COPY requirements.txt .
ENV HF_HUB_ENABLE_HF_TRANSFER=1
RUN pip install --upgrade pip "setuptools<70.0.0" wheel
# TODO, unpin setuptools when this issue in flash attention is resolved
RUN pip install torch==2.3.0 torchvision==0.18.0 torchaudio==2.3.0 --index-url https://download.pytorch.org/whl/cu121
Expand Down

0 comments on commit e77830e

Please sign in to comment.