Move ONNX runtime into general requirements.txt from cpu-only Dockerfile

remsky · Jan 9, 2025 · f6e3afa · f6e3afa
1 parent 65d0773
commit f6e3afa
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/Dockerfile.cpu b/Dockerfile.cpu
@@ -11,8 +11,7 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
     && rm -rf /var/lib/apt/lists/*
 
 # Install PyTorch CPU version and ONNX runtime
-RUN pip3 install --no-cache-dir torch==2.5.1 --extra-index-url https://download.pytorch.org/whl/cpu && \
-    pip3 install --no-cache-dir onnxruntime==1.20.1
+RUN pip3 install --no-cache-dir torch==2.5.1 --extra-index-url https://download.pytorch.org/whl/cpu 
 
 # Install all other dependencies from requirements.txt
 COPY requirements.txt .

diff --git a/requirements.txt b/requirements.txt
@@ -11,6 +11,7 @@ sqlalchemy==2.0.27
 transformers==4.47.1
 numpy==2.2.1
 scipy==1.14.1
+onnxruntime==1.20.1
 
 # Audio processing
 soundfile==0.13.0