423A35C7/LLaMA-Factory
1
0
Fork 0
mirror of https://github.com/hiyouga/LLaMA-Factory.git synced 2025-12-14 19:06:26 +08:00
Code Issues Packages Projects Releases Wiki Activity

Update Dockerfile

Browse Source
This commit is contained in:
hoshi-hiyouga
2024-06-24 23:41:35 +08:00
committed by GitHub
parent d7207e8ad1
commit dcc2e24f5c
1 changed files with 4 additions and 6 deletions
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 ARG PIP_INDEX=https://pypi.org/simple
# Set the working directory # Set the working directory
WORKDIR /app/LLaMA-Factory WORKDIR /app
RUN cd /app && \
git config --global http.version HTTP/1.1 && \
git clone https://github.com/hiyouga/LLaMA-Factory.git && \
cd /app/LLaMA-Factory
# Install the requirements # Install the requirements
COPY requirements.txt /app/
RUN pip config set global.index-url $PIP_INDEX RUN pip config set global.index-url $PIP_INDEX
RUN python -m pip install --upgrade pip RUN python -m pip install --upgrade pip
RUN python -m pip install -r requirements.txt RUN python -m pip install -r requirements.txt
# Copy the rest of the application into the image
COPY . /app/
# Install the LLaMA Factory # Install the LLaMA Factory
RUN EXTRA_PACKAGES="metrics"; \ RUN EXTRA_PACKAGES="metrics"; \