LLaMA-Factory/Dockerfile

15 lines
296 B
Docker
Raw Normal View History

FROM nvcr.io/nvidia/pytorch:24.01-py3
WORKDIR /app
COPY requirements.txt /app/
RUN pip install -r requirements.txt
COPY . /app/
2024-05-16 10:39:08 +00:00
RUN pip install -e .[metrics,bitsandbytes,qwen]
VOLUME [ "/root/.cache/huggingface/", "/app/data", "/app/output" ]
EXPOSE 7860
2024-05-07 06:09:48 +00:00
CMD [ "llamafactory-cli", "webui" ]