LLaMA-Factory/Dockerfile
junwooo.lee a274f7d069 fix: splitted Dockerfile's CMD
Former-commit-id: 4598734a0dc837be5f30033fb16d22b6a4d38913
2024-05-07 15:09:48 +09:00

15 lines
306 B
Docker

FROM nvcr.io/nvidia/pytorch:24.01-py3
WORKDIR /app
COPY requirements.txt /app/
RUN pip install -r requirements.txt
COPY . /app/
RUN pip install -e .[deepspeed,metrics,bitsandbytes,qwen]
VOLUME [ "/root/.cache/huggingface/", "/app/data", "/app/output" ]
EXPOSE 7860
CMD [ "llamafactory-cli", "webui" ]