Skip to content

Commit 3252d7a

Browse files
authored
unpin torch versions for onnx Dockerfile (#11290)
unpin torch versions for onnx
1 parent b316104 commit 3252d7a

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

docker/diffusers-onnxruntime-cpu/Dockerfile

+3-3
Original file line numberDiff line numberDiff line change
@@ -28,9 +28,9 @@ ENV PATH="/opt/venv/bin:$PATH"
2828
# pre-install the heavy dependencies (these can later be overridden by the deps from setup.py)
2929
RUN python3 -m pip install --no-cache-dir --upgrade pip uv==0.1.11 && \
3030
python3 -m uv pip install --no-cache-dir \
31-
torch==2.1.2 \
32-
torchvision==0.16.2 \
33-
torchaudio==2.1.2 \
31+
torch \
32+
torchvision \
33+
torchaudio\
3434
onnxruntime \
3535
--extra-index-url https://download.pytorch.org/whl/cpu && \
3636
python3 -m uv pip install --no-cache-dir \

0 commit comments

Comments
 (0)