FROM us-docker.pkg.dev/deeplearning-platform-release/gcr.io/base-cu124.py310

WORKDIR /app
COPY requirements.txt requirements.txt
RUN pip install -r requirements.txt

COPY src src
COPY configs configs
COPY llm_finetune.py llm_finetune.py
COPY run_llm.py run_llm.py
