423A35C7/LLaMA-Factory
1
0
Fork 0
mirror of https://github.com/hiyouga/LLaMA-Factory.git synced 2025-11-04 18:02:19 +08:00
Code Issues Packages Projects Releases Wiki Activity

Update Dockerfile

Browse Source 
Former-commit-id: 632681d8ece0eaac59bb364d971435a3bc6665a9
This commit is contained in:
hoshi-hiyouga 2024-06-24 23:41:35 +08:00 committed by GitHub
parent af2607de1a
commit 2e5d521ed8
1 changed files with 4 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
docker/docker-cuda/Dockerfile
View File

@ -9,18 +9,16 @@ ARG INSTALL_DEEPSPEED=false
ARG PIP_INDEX=https://pypi.org/simple
# Set the working directory
WORKDIR /app/LLaMA-Factory
RUN cd /app && \
git config --global http.version HTTP/1.1 && \
git clone https://github.com/hiyouga/LLaMA-Factory.git && \
cd /app/LLaMA-Factory
WORKDIR /app
# Install the requirements
COPY requirements.txt /app/
RUN pip config set global.index-url $PIP_INDEX
RUN python -m pip install --upgrade pip
RUN python -m pip install -r requirements.txt
# Copy the rest of the application into the image
COPY . /app/
# Install the LLaMA Factory
RUN EXTRA_PACKAGES="metrics"; \