update the requirements.txt for tts and asr (opea-project#550)

Signed-off-by: chensuyue <[email protected]>
predictionguard · Aug 22, 2024 · 5ba2561 · 5ba2561
1 parent 97fdf54
commit 5ba2561
Show file tree

Hide file tree

Showing 6 changed files with 4 additions and 6 deletions.
diff --git a/comps/asr/Dockerfile b/comps/asr/Dockerfile
@@ -15,6 +15,7 @@ COPY comps /home/user/comps
 
 RUN pip install --no-cache-dir --upgrade pip && \
     if [ "${ARCH}" = "cpu" ]; then \
+        pip install torch torchvision --index-url https://download.pytorch.org/whl/cpu ; \
         pip install --no-cache-dir --extra-index-url https://download.pytorch.org/whl/cpu -r /home/user/comps/asr/requirements.txt ; \
     else \
         pip install --no-cache-dir -r /home/user/comps/asr/requirements.txt ; \

diff --git a/comps/asr/requirements.txt b/comps/asr/requirements.txt
@@ -4,13 +4,10 @@ fastapi
 opentelemetry-api
 opentelemetry-exporter-otlp
 opentelemetry-sdk
-optimum[habana]
 prometheus-fastapi-instrumentator
 pydantic==2.7.2
 pydub
 shortuuid
-torch
-torchvision
 transformers
 uvicorn
 zhconv
diff --git a/comps/asr/whisper/Dockerfile b/comps/asr/whisper/Dockerfile
@@ -23,6 +23,7 @@ USER user
 RUN pip install --no-cache-dir --upgrade pip && \
     pip install --no-cache-dir -r /home/user/comps/asr/requirements.txt && \
     if [ "${ARCH}" = "cpu" ]; then \
+        pip install torch torchvision --index-url https://download.pytorch.org/whl/cpu ; \
         pip install --no-cache-dir --extra-index-url https://download.pytorch.org/whl/cpu -r /home/user/comps/asr/requirements.txt ; \
     else \
         pip install --no-cache-dir -r /home/user/comps/asr/requirements.txt ; \

diff --git a/comps/tts/Dockerfile b/comps/tts/Dockerfile
@@ -13,6 +13,7 @@ COPY comps /home/user/comps
 
 RUN pip install --no-cache-dir --upgrade pip && \
     if [ "${ARCH}" = "cpu" ]; then \
+        pip install torch torchvision --index-url https://download.pytorch.org/whl/cpu ; \
         pip install --no-cache-dir --extra-index-url https://download.pytorch.org/whl/cpu -r /home/user/comps/tts/requirements.txt ; \
     else \
         pip install --no-cache-dir -r /home/user/comps/tts/requirements.txt ; \

diff --git a/comps/tts/requirements.txt b/comps/tts/requirements.txt
@@ -4,12 +4,9 @@ fastapi
 opentelemetry-api
 opentelemetry-exporter-otlp
 opentelemetry-sdk
-optimum[habana]
 prometheus-fastapi-instrumentator
 sentencepiece
 shortuuid
 soundfile
-torch
-torchvision
 transformers
 uvicorn
diff --git a/comps/tts/speecht5/Dockerfile b/comps/tts/speecht5/Dockerfile
@@ -22,6 +22,7 @@ USER user
 
 RUN pip install --no-cache-dir --upgrade pip && \
     if [ "${ARCH}" = "cpu" ]; then \
+        pip install torch torchvision --index-url https://download.pytorch.org/whl/cpu ; \
         pip install --no-cache-dir --extra-index-url https://download.pytorch.org/whl/cpu -r /home/user/comps/tts/requirements.txt ; \
     else \
         pip install --no-cache-dir -r /home/user/comps/tts/requirements.txt ; \