m

handler.py

@@ -1,10 +1,14 @@
 import string
 import warnings
 warnings.filterwarnings('ignore')
+import subprocess, io, os, sys, time
+
+os.environ["XFORMERS_DISABLE_FLASH_ATTN"] = "1"
+result = subprocess.run(['pip', 'install', 'git+https://github.com/facebookresearch/xformers.git@main#egg=xformers'], check=True)
+
 from entklei import get_nude
 from scipy.ndimage import binary_dilation
 
-import subprocess, io, os, sys, time
 
 is_production = True
 install_stuff = True

requirements.txt

@@ -46,7 +46,7 @@ tencentcloud-sdk-python
 pydash
 boto3
 ninja  # (Optional) Makes the build much faster
-git+https://github.com/facebookresearch/xformers.git@main#egg=xformers
+# git+https://github.com/facebookresearch/xformers.git@main#egg=xformers
 git+https://github.com/huggingface/diffusers.git@main#egg=diffusers
 git+https://github.com/huggingface/peft.git@main#egg=peft
 git+https://github.com/huggingface/transformers.git@main#egg=transformers