We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent b316104 commit 3252d7aCopy full SHA for 3252d7a
docker/diffusers-onnxruntime-cpu/Dockerfile
@@ -28,9 +28,9 @@ ENV PATH="/opt/venv/bin:$PATH"
28
# pre-install the heavy dependencies (these can later be overridden by the deps from setup.py)
29
RUN python3 -m pip install --no-cache-dir --upgrade pip uv==0.1.11 && \
30
python3 -m uv pip install --no-cache-dir \
31
- torch==2.1.2 \
32
- torchvision==0.16.2 \
33
- torchaudio==2.1.2 \
+ torch \
+ torchvision \
+ torchaudio\
34
onnxruntime \
35
--extra-index-url https://download.pytorch.org/whl/cpu && \
36
0 commit comments