diff --git a/docker/docker-npu/Dockerfile b/docker/docker-npu/Dockerfile index d4b879e2..c75997c1 100644 --- a/docker/docker-npu/Dockerfile +++ b/docker/docker-npu/Dockerfile @@ -16,6 +16,8 @@ ARG TORCH_INDEX=https://download.pytorch.org/whl/cpu # Set the working directory WORKDIR /app +RUN apt-get update && apt-get install -y git + # Install the requirements COPY requirements.txt /app RUN pip config set global.index-url "$PIP_INDEX" && \ diff --git a/requirements.txt b/requirements.txt index 93e83530..96a0714a 100644 --- a/requirements.txt +++ b/requirements.txt @@ -19,3 +19,4 @@ fire packaging pyyaml numpy<2.0.0 +modelscope