fix(diffusers): use nightly rocm for hipblas builds

Signed-off-by: Ettore Di Giacinto <[email protected]>
mudler · Aug 8, 2024 · e0f8c9c · e0f8c9c
1 parent 4a1a3a5
commit e0f8c9c
Showing 1 changed file with 3 additions and 2 deletions.
diff --git 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
+sentencepiece