p12475398
/
LLaMA-Factory-Mirror
forked from p04798526/LLaMA-Factory-Mirror
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity
LLaMA-Factory-Mirror/Dockerfile

15 lines
328 B
Docker
Raw Normal View History

Add dockerize support Already tested with the model of Qwen:1.8B and the dataset of alpaca_data_zh. Some python libraries are added to the Dockerfile as a result of the exception messages displayed throughout test procedure.
2024-03-08 10:47:28 +08:00
FROM cnstark/pytorch:2.0.1-py3.9.17-cuda11.8.0-ubuntu20.04
WORKDIR /app
COPY requirements.txt /app/
improve Docker build and runtime parameters Modify installation method of extra python library. Utilize shared memory of the host machine to increase training performance.
2024-03-12 14:05:10 +08:00
RUN pip install -r requirements.txt
Add dockerize support Already tested with the model of Qwen:1.8B and the dataset of alpaca_data_zh. Some python libraries are added to the Dockerfile as a result of the exception messages displayed throughout test procedure.
2024-03-08 10:47:28 +08:00
COPY . /app/
improve Docker build and runtime parameters Modify installation method of extra python library. Utilize shared memory of the host machine to increase training performance.
2024-03-12 14:05:10 +08:00
RUN pip install -e .[deepspeed,metrics,bitsandbytes,qwen]
Add dockerize support Already tested with the model of Qwen:1.8B and the dataset of alpaca_data_zh. Some python libraries are added to the Dockerfile as a result of the exception messages displayed throughout test procedure.
2024-03-08 10:47:28 +08:00
VOLUME [ "/root/.cache/huggingface/", "/app/data", "/app/output" ]
EXPOSE 7860
CMD [ "python", "src/train_web.py" ]