FROM nvcr.io/nvidia/pytorch:24.04-py3

RUN git clone https://github.com/muellerzr/llama-3-8b-self-align/ \
    && cd llama-3-8b-self-align \
    && git clone https://github.com/tree-sitter/tree-sitter-python --depth=1 \
    && pip install -r requirements.txt \
    && pip install datasets vllm

CMD ["/bin/bash"]