diff --git a/docker/docker-cuda/Dockerfile b/docker/docker-cuda/Dockerfile index 72797a26..61d58005 100644 --- a/docker/docker-cuda/Dockerfile +++ b/docker/docker-cuda/Dockerfile @@ -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"; \