22 lines
486 B
Docker
22 lines
486 B
Docker
ARG BASE_TAG=main-base
|
|
FROM winglian/axolotl-base:$BASE_TAG
|
|
|
|
ARG TORCH_CUDA_ARCH_LIST="7.0 7.5 8.0 8.6+PTX"
|
|
|
|
RUN apt-get update && \
|
|
apt-get install -y vim curl
|
|
|
|
WORKDIR /workspace
|
|
|
|
# The base image ships with `pydantic==1.8.2` which is not working
|
|
RUN python3 -m pip install -U --no-cache-dir pydantic
|
|
|
|
ARG REF=main
|
|
RUN mkdir axolotl
|
|
COPY . axolotl/
|
|
RUN cd axolotl && \
|
|
pip install -e .[int4]
|
|
|
|
# helper for huggingface-login cli
|
|
RUN git config --global credential.helper store
|