Dockerfile

FROM nroggendorff/train-llama:latest

USER user
RUN python -u prep.py

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