Dockerfile

FROM nroggendorff/train-llama:latest
RUN python -u prep.py

CMD ["python", "train.py"]