Skip to content

Commit f6e3afa

Browse files
committed
Move ONNX runtime into general requirements.txt from cpu-only Dockerfile
1 parent 65d0773 commit f6e3afa

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

Dockerfile.cpu

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -11,8 +11,7 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
1111
&& rm -rf /var/lib/apt/lists/*
1212

1313
# Install PyTorch CPU version and ONNX runtime
14-
RUN pip3 install --no-cache-dir torch==2.5.1 --extra-index-url https://download.pytorch.org/whl/cpu && \
15-
pip3 install --no-cache-dir onnxruntime==1.20.1
14+
RUN pip3 install --no-cache-dir torch==2.5.1 --extra-index-url https://download.pytorch.org/whl/cpu
1615

1716
# Install all other dependencies from requirements.txt
1817
COPY requirements.txt .

requirements.txt

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -11,6 +11,7 @@ sqlalchemy==2.0.27
1111
transformers==4.47.1
1212
numpy==2.2.1
1313
scipy==1.14.1
14+
onnxruntime==1.20.1
1415

1516
# Audio processing
1617
soundfile==0.13.0

0 commit comments

Comments
 (0)