lnyan commited on
Commit
cb7f9c8
1 Parent(s): dafd8ad

Disable xformer

Browse files
Files changed (2) hide show
  1. README.md +4 -4
  2. app.py +2 -2
README.md CHANGED
@@ -1,12 +1,12 @@
1
  ---
2
  title: Stablediffusion Infinity
3
- emoji: 🐨
4
- colorFrom: green
5
  colorTo: purple
6
  sdk: gradio
7
- sdk_version: 3.4
8
  app_file: app.py
9
- pinned: false
10
  license: apache-2.0
11
  ---
12
 
 
1
  ---
2
  title: Stablediffusion Infinity
3
+ emoji: ♾️
4
+ colorFrom: blue
5
  colorTo: purple
6
  sdk: gradio
7
+ sdk_version: 3.10.1
8
  app_file: app.py
9
+ pinned: true
10
  license: apache-2.0
11
  ---
12
 
app.py CHANGED
@@ -1,7 +1,7 @@
1
  import subprocess
2
  # import os.path as osp
3
  import pip
4
- pip.main(["install","-v","-U","git+https://github.com/facebookresearch/xformers.git@main#egg=xformers"])
5
  # subprocess.check_call("pip install -v -U git+https://github.com/facebookresearch/xformers.git@main#egg=xformers", cwd=osp.dirname(__file__), shell=True)
6
 
7
  import io
@@ -308,7 +308,7 @@ class StableDiffusionInpaint:
308
  inpaint.tokenizer,
309
  )
310
  inpaint.to(device)
311
- inpaint.enable_xformers_memory_efficient_attention()
312
  # if device == "mps":
313
  # _ = text2img("", num_inference_steps=1)
314
  scheduler_dict["PLMS"] = inpaint.scheduler
 
1
  import subprocess
2
  # import os.path as osp
3
  import pip
4
+ # pip.main(["install","-v","-U","git+https://github.com/facebookresearch/xformers.git@main#egg=xformers"])
5
  # subprocess.check_call("pip install -v -U git+https://github.com/facebookresearch/xformers.git@main#egg=xformers", cwd=osp.dirname(__file__), shell=True)
6
 
7
  import io
 
308
  inpaint.tokenizer,
309
  )
310
  inpaint.to(device)
311
+ # inpaint.enable_xformers_memory_efficient_attention()
312
  # if device == "mps":
313
  # _ = text2img("", num_inference_steps=1)
314
  scheduler_dict["PLMS"] = inpaint.scheduler