public-mirrors/Kokoro-FastAPI
1
0
Fork 0
mirror of https://github.com/remsky/Kokoro-FastAPI.git synced 2025-04-13 09:39:17 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

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

Browse source
This commit is contained in:
remsky 2025-01-09 13:11:04 -07:00
parent 65d0773e1e
commit f6e3afa14c
2 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
Dockerfile.cpu
View file

@ -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 .

1
requirements.txt
View file

@ -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