|
|
|
FROM gcr.io/ai2-beaker-core/public/cjvktq5s0r0fr8pb7470:latest |
|
|
|
RUN apt update && apt install -y openjdk-8-jre-headless |
|
|
|
RUN curl -s https://packagecloud.io/install/repositories/github/git-lfs/script.deb.sh | bash |
|
RUN apt-get -y install git-lfs |
|
|
|
WORKDIR /stage/ |
|
|
|
COPY requirements.txt . |
|
RUN pip install --upgrade pip setuptools wheel |
|
RUN pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118 |
|
RUN pip install packaging |
|
RUN pip install flash-attn --no-build-isolation |
|
RUN pip install -r requirements.txt |
|
|
|
COPY open_instruct open_instruct |
|
COPY eval eval |
|
COPY ds_configs ds_configs |
|
COPY scripts scripts |
|
RUN chmod +x scripts/* |
|
|
|
|
|
RUN chmod -R 777 /stage/ |