diff --git a/backend/python/diffusers/requirements-hipblas.txt b/backend/python/diffusers/requirements-hipblas.txt index 17cf7249..838a045b 100644 --- a/backend/python/diffusers/requirements-hipblas.txt +++ b/backend/python/diffusers/requirements-hipblas.txt @@ -1,6 +1,6 @@ --extra-index-url https://download.pytorch.org/whl/rocm6.0 torch==2.3.1+rocm6.0 -torchvision==0.18.1+rocm6.0 +torchvision==0.21.0+cu118 diffusers opencv-python transformers diff --git a/backend/python/diffusers/requirements-intel.txt b/backend/python/diffusers/requirements-intel.txt index eb7448b0..7a90bde3 100644 --- a/backend/python/diffusers/requirements-intel.txt +++ b/backend/python/diffusers/requirements-intel.txt @@ -1,7 +1,7 @@ --extra-index-url https://pytorch-extension.intel.com/release-whl/stable/xpu/us/ intel-extension-for-pytorch==2.3.110+xpu torch==2.3.1+cxx11.abi -torchvision==0.18.1+cxx11.abi +torchvision==0.21.0+cu118 oneccl_bind_pt==2.3.100+xpu optimum[openvino] setuptools