Skip to content

Commit 85bdc1d

Browse files
authored
Fix PSIRT Vulnerability - Dependency Confusion in oneccl_bind_pt package (#92)
* Fix PRIST * Fix PRIST
1 parent 58baf8f commit 85bdc1d

File tree

2 files changed

+4
-2
lines changed

2 files changed

+4
-2
lines changed

visual-ai/ComfyUI/docker/Dockerfile

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -21,7 +21,8 @@ RUN wget -O- https://apt.repos.intel.com/intel-gpg-keys/GPG-PUB-KEY-INTEL-SW-PRO
2121
curl -sS https://bootstrap.pypa.io/get-pip.py | python3.10 && \
2222
update-alternatives --install /usr/bin/python3 python3 /usr/bin/python3.10 1 && \
2323
pip install torch==2.7.0 torchvision==0.22.0 torchaudio==2.7.0 --index-url https://download.pytorch.org/whl/xpu && \
24-
pip install intel-extension-for-pytorch==2.7.10+xpu oneccl_bind_pt==2.7.0+xpu --extra-index-url https://pytorch-extension.intel.com/release-whl/stable/xpu/cn/ && \
24+
pip install intel-extension-for-pytorch==2.7.10+xpu --extra-index-url https://pytorch-extension.intel.com/release-whl/stable/xpu/cn/ && \
25+
pip install oneccl_bind_pt==2.7.0+xpu --index-url https://pytorch-extension.intel.com/release-whl/stable/xpu/cn/ && \
2526
pip install bigdl-core-xe-all==2.6.0 --extra-index-url https://download.pytorch.org/whl/xpu && \
2627
apt remove python3-blinker -y && \
2728
# Install xDit related dependencies

visual-ai/Wan2.2/docker/Dockerfile

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -93,7 +93,8 @@ CMD ["bash", "-c", "source /root/.bashrc && exec bash"]
9393
ENV LD_LIBRARY_PATH="/usr/local/lib:/usr/local/lib/python3.10/dist-packages/torch/lib:$LD_LIBRARY_PATH"
9494

9595
RUN pip install torch==2.7.0 torchvision==0.22.0 torchaudio==2.7.0 --index-url https://download.pytorch.org/whl/xpu && \
96-
pip install intel-extension-for-pytorch==2.7.10+xpu oneccl_bind_pt==2.7.0+xpu --extra-index-url https://pytorch-extension.intel.com/release-whl/stable/xpu/cn/ && \
96+
pip install intel-extension-for-pytorch==2.7.10+xpu --extra-index-url https://pytorch-extension.intel.com/release-whl/stable/xpu/cn/ && \
97+
pip install oneccl_bind_pt==2.7.0+xpu --index-url https://pytorch-extension.intel.com/release-whl/stable/xpu/cn/ && \
9798
pip install bigdl-core-xe-all==2.6.0 --extra-index-url https://download.pytorch.org/whl/xpu && \
9899
apt remove python3-blinker -y
99100

0 commit comments

Comments
 (0)