From 2e2a0dffbc4aae5ad1225278d98370a6ec898657 Mon Sep 17 00:00:00 2001 From: Ettore Di Giacinto Date: Fri, 9 Aug 2024 22:36:10 +0200 Subject: [PATCH] fix(diffusers-hipblas): pin to rocm6.1 As per https://rocm.docs.amd.com/projects/install-on-linux/en/latest/install/3rd-party/pytorch-install.html Signed-off-by: Ettore Di Giacinto --- backend/python/diffusers/requirements-hipblas.txt | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/backend/python/diffusers/requirements-hipblas.txt b/backend/python/diffusers/requirements-hipblas.txt index b7890f6e..8c4c070c 100644 --- a/backend/python/diffusers/requirements-hipblas.txt +++ b/backend/python/diffusers/requirements-hipblas.txt @@ -1,4 +1,4 @@ ---extra-index-url https://download.pytorch.org/whl/nightly/ +--extra-index-url https://download.pytorch.org/whl/nightly/rocm6.1/ torch torchvision diffusers