tocmo0nlord/axolotl
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

Revert "run PR e2e docker CI tests in Modal" (#1220) [skip ci]

Browse Source
This commit is contained in:
Wing Lian
2024-01-26 16:50:44 -05:00
committed by GitHub
parent 36d053f6f0
commit 8da1633124
6 changed files with 42 additions and 141 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
docker/Dockerfile-modal → docker/Dockerfile-tests
View File

@@ -1,11 +1,14 @@
FROM winglian/axolotl-base:main-base
ARG BASE_TAG=main-base
FROM winglian/axolotl-base:$BASE_TAG
ENV TORCH_CUDA_ARCH_LIST="7.0 7.5 8.0 8.6+PTX"
ENV AXOLOTL_EXTRAS=""
ENV CUDA="118"
ENV BNB_CUDA_VERSION="118"
ENV PYTORCH_VERSION="2.0.1"
ENV GITHUB_REF="main"
ARG TORCH_CUDA_ARCH_LIST="7.0 7.5 8.0 8.6+PTX"
ARG AXOLOTL_EXTRAS=""
ARG CUDA="118"
ENV BNB_CUDA_VERSION=$CUDA
ARG PYTORCH_VERSION="2.0.1"
ARG GITHUB_REF="main"
ENV PYTORCH_VERSION=$PYTORCH_VERSION
RUN apt-get update && \
apt-get install -y --allow-change-held-packages vim curl nano libnccl2 libnccl-dev