Skip to content

GitLab

Unverified Commit 3252d7ad authored by Sayak Paul's avatar Sayak Paul Committed by GitHub
Browse files

unpin torch versions for onnx Dockerfile (#11290) 

unpin torch versions for onnx
parent b316104d
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 3 deletions
docker/diffusers-onnxruntime-cpu/Dockerfile
View file @ 3252d7ad
...@@ -28,9 +28,9 @@ ENV PATH="/opt/venv/bin:$PATH" ...@@ -28,9 +28,9 @@ ENV PATH="/opt/venv/bin:$PATH"
# pre-install the heavy dependencies (these can later be overridden by the deps from setup.py) # pre-install the heavy dependencies (these can later be overridden by the deps from setup.py)
RUN python3 -m pip install --no-cache-dir --upgrade pip uv==0.1.11 && \ RUN python3 -m pip install --no-cache-dir --upgrade pip uv==0.1.11 && \
python3 -m uv pip install --no-cache-dir \ python3 -m uv pip install --no-cache-dir \
torch==2.1.2 \ torch \
torchvision==0.16.2 \ torchvision \
torchaudio==2.1.2 \ torchaudio\
onnxruntime \ onnxruntime \
--extra-index-url https://download.pytorch.org/whl/cpu && \ --extra-index-url https://download.pytorch.org/whl/cpu && \
python3 -m uv pip install --no-cache-dir \ python3 -m uv pip install --no-cache-dir \
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment