From fac5e83af93c3826f15b138bbf838c850909fa8a Mon Sep 17 00:00:00 2001
From: wql <wengqinlan18@163.com>
Date: Fri, 13 Sep 2024 10:53:23 +0800
Subject: [PATCH] chore: add auto install

---
 docker/docker-npu/Dockerfile | 2 ++
 requirements.txt             | 1 +
 2 files changed, 3 insertions(+)

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