Add flash attention

1861450d · novelailab · 30541064 · 1861450d · 295d61e5
Commit 1861450d authored Sep 29, 2022 by novelailab
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 1 deletion

Dockerfile Dockerfile +1 -0

stable-diffusion-private stable-diffusion-private +1 -1

No files found.
--- a/Dockerfile
+++ b/Dockerfile
@@ -33,6 +33,7 @@ RUN pip3 install https://www.dropbox.com/s/8ozhhbo1g7y5dsz/basedformer-f7c8c4fe1
 #built DS
 RUN pip3 install https://www.dropbox.com/s/euzpgpfrs9isf1z/deepspeed-0.7.3%2B55b7b9e0-cp38-cp38-linux_x86_64.whl?dl=1
 RUN pip3 install faiss-cpu sentence_transformers
+RUN pip3 install 'git+https://github.com/facebookresearch/xformers@51dd119#egg=xformers'

 #Open ports
 EXPOSE 8080

--- a/stable-diffusion-private @ 295d61e5
+++ b/stable-diffusion-private @ 295d61e5
-Subproject commit 4c05356ddda54187b8af61ae27a118fb6550c777
+Subproject commit 295d61e5c36103b4f1c566be801d89880280ea10