Spaces:

lnyan
/

stablediffusion-infinity

Runtime error

lnyan commited on Nov 27, 2022

Commit

98778ff

•

1 Parent(s): 59868ce

Add xformers

Files changed (2) hide show

app.py CHANGED Viewed

@@ -302,6 +302,7 @@ class StableDiffusionInpaint:
                         inpaint.tokenizer,
                     )
         inpaint.to(device)
         # if device == "mps":
         # _ = text2img("", num_inference_steps=1)
         scheduler_dict["PLMS"] = inpaint.scheduler

                         inpaint.tokenizer,
                     )
         inpaint.to(device)
+        inpaint.enable_xformers_memory_efficient_attention()
         # if device == "mps":
         # _ = text2img("", num_inference_steps=1)
         scheduler_dict["PLMS"] = inpaint.scheduler

requirements.txt CHANGED Viewed

@@ -12,4 +12,6 @@ diffusers==0.9.0
 transformers
 ftfy
 fpie
-accelerate

 transformers
 ftfy
 fpie
+accelerate
+ninja
+-e git+https://github.com/facebookresearch/xformers.git@main#egg=xformers