From 8317839ca5e36fe6abcb981ce26f0ea6428fa7d2 Mon Sep 17 00:00:00 2001 From: Ettore Di Giacinto Date: Thu, 8 Aug 2024 17:28:07 +0200 Subject: [PATCH] fix(diffusers): use nightly rocm for hipblas builds (#3202) Signed-off-by: Ettore Di Giacinto --- backend/python/diffusers/requirements-hipblas.txt | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/backend/python/diffusers/requirements-hipblas.txt b/backend/python/diffusers/requirements-hipblas.txt index 9e992d02b907..92987e7aeb99 100644 --- a/backend/python/diffusers/requirements-hipblas.txt +++ b/backend/python/diffusers/requirements-hipblas.txt @@ -1,4 +1,5 @@ ---extra-index-url https://download.pytorch.org/whl/rocm6.0 +--pre +--extra-index-url https://download.pytorch.org/whl/nightly/ torch torchvision diffusers @@ -7,4 +8,4 @@ transformers accelerate compel peft -sentencepiece \ No newline at end of file +sentencepiece