From d72fe4456d91be55c16d157d3b658c1c5cb6e2ab Mon Sep 17 00:00:00 2001 From: Andrew Powers-Holmes Date: Mon, 26 Feb 2024 16:43:35 +1100 Subject: [PATCH] allow prerelease xformers --- docker/base/Dockerfile | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/docker/base/Dockerfile b/docker/base/Dockerfile index e6c3277..7f6b3c6 100644 --- a/docker/base/Dockerfile +++ b/docker/base/Dockerfile @@ -167,8 +167,8 @@ FROM base as xformers-binary ARG XFORMERS_PACKAGE="xformers" ARG XFORMERS_PIP_ARGS="" RUN --mount=type=cache,target=/root/.cache/pip,sharing=locked \ - python -m pip install ${XFORMERS_PIP_ARGS} "${XFORMERS_PACKAGE}" - + python -m pip install ${XFORMERS_PIP_ARGS} "${XFORMERS_PACKAGE}" \ + || python -m pip install --pre ${XFORMERS_PIP_ARGS} "${XFORMERS_PACKAGE}" # or if xformers has to be compiled; flash attention 2 makes this impossible # on a public github actions runner without limiting yourself to only a handful of architectures