fix(diffusers): allow pre-releases for requirements

Signed-off-by: Ettore Di Giacinto <[email protected]>
dave-gray101 · Aug 9, 2024 · 5fcafc3 · 5fcafc3
1 parent 74f8785
commit 5fcafc3
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/backend/python/diffusers/install.sh b/backend/python/diffusers/install.sh
@@ -11,4 +11,9 @@ if [ "x${BUILD_PROFILE}" == "xintel" ]; then
     EXTRA_PIP_INSTALL_FLAGS+=" --upgrade --index-strategy=unsafe-first-match"
 fi
 
+# hipblas builds from nightly that needs pre-releases to work
+if [ "x${BUILD_PROFILE}" == "xhipblas" ]; then
+    EXTRA_PIP_INSTALL_FLAGS+=" --prerelease=allow"
+fi
+
 installRequirements