multimodalart HF staff commited on
Commit
af17881
1 Parent(s): f187f09

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -4,7 +4,7 @@ import os
4
  gpu_info = getoutput('nvidia-smi')
5
  if("A10G" in gpu_info):
6
  which_gpu = "A10G"
7
- os.system(f"pip install -q https://github.com/camenduru/stable-diffusion-webui-colab/releases/download/0.0.15/xformers-0.0.15.dev0+4c06c79.d20221205-cp38-cp38-linux_x86_64.whl")
8
  elif("T4" in gpu_info):
9
  which_gpu = "T4"
10
  os.system(f"pip install -q https://github.com/camenduru/stable-diffusion-webui-colab/releases/download/0.0.15/xformers-0.0.15.dev0+1515f77.d20221130-cp38-cp38-linux_x86_64.whl")
 
4
  gpu_info = getoutput('nvidia-smi')
5
  if("A10G" in gpu_info):
6
  which_gpu = "A10G"
7
+ os.system(f"pip install --no-deps xformers==0.0.16rc425")
8
  elif("T4" in gpu_info):
9
  which_gpu = "T4"
10
  os.system(f"pip install -q https://github.com/camenduru/stable-diffusion-webui-colab/releases/download/0.0.15/xformers-0.0.15.dev0+1515f77.d20221130-cp38-cp38-linux_x86_64.whl")