train-llama / Dockerfile
nroggendorff's picture
Update Dockerfile
be2f96f verified
raw
history blame
219 Bytes
FROM python:latest
WORKDIR /app
COPY . .
RUN pip install --no-cache-dir --upgrade -r requirements.txt
RUN mkdir /.cache && chmod 777 /.cache
RUN mkdir /app/model && chmod 777 /app/model
CMD ["python3", "train.py"]