Merge pull request #124 from qslug/fix-xformers-again

Find last known working xformers commit
This commit is contained in:
Victor Hall 2023-03-27 12:12:02 -04:00 committed by GitHub
commit 7df4606b26
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions

View File

@ -30,7 +30,7 @@ RUN --mount=type=cache,target=/root/.cache/pip \
pip install -U -I torch==1.13.1+cu117 torchvision==0.14.1+cu117 --extra-index-url "https://download.pytorch.org/whl/cu117" && \
pip install -r requirements.txt && \
export FORCE_CUDA=1 && export TORCH_CUDA_ARCH_LIST="7.5;8.0;8.6" && export CUDA_VISIBLE_DEVICES=0 && \
pip install --no-deps git+https://github.com/facebookresearch/xformers.git@48a77cc#egg=xformers
pip install --no-deps git+https://github.com/facebookresearch/xformers.git@e14dba4#egg=xformers
###################