Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
renzhc
diffusers_dcu
Commits
3252d7ad
Unverified
Commit
3252d7ad
authored
Apr 16, 2025
by
Sayak Paul
Committed by
GitHub
Apr 16, 2025
Browse files
unpin torch versions for onnx Dockerfile (#11290)
unpin torch versions for onnx
parent
b316104d
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
3 additions
and
3 deletions
+3
-3
docker/diffusers-onnxruntime-cpu/Dockerfile
docker/diffusers-onnxruntime-cpu/Dockerfile
+3
-3
No files found.
docker/diffusers-onnxruntime-cpu/Dockerfile
View file @
3252d7ad
...
...
@@ -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)
RUN
python3
-m
pip
install
--no-cache-dir
--upgrade
pip
uv
==
0.1.11
&&
\
python3
-m
uv pip
install
--no-cache-dir
\
torch
==
2.1.2
\
torchvision
==
0.16.2
\
torchaudio
==
2.1.2
\
torch
\
torchvision
\
torchaudio
\
onnxruntime
\
--extra-index-url
https://download.pytorch.org/whl/cpu
&&
\
python3
-m
uv pip
install
--no-cache-dir
\
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment