Hugging Face's logo

Spaces:
darkstorm2150
/
protogen-web-ui
Running on A10G

App Files Community
29
darkstorm2150 commited on
Commit
f4cab5f
•
1 Parent(s): 4c661d1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -1,11 +1,13 @@
1
  import os
2
  from subprocess import getoutput
3
 
 
 
4
  gpu_info = getoutput('nvidia-smi')
5
  if("A10G" in gpu_info):
6
- 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")
7
  elif("T4" in gpu_info):
8
- 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")
9
 
10
  os.system(f"git clone -b v1.5 https://github.com/camenduru/stable-diffusion-webui /home/user/app/stable-diffusion-webui")
11
  os.chdir("/home/user/app/stable-diffusion-webui")
 
1
  import os
2
  from subprocess import getoutput
3
 
4
+ pip install --upgrade pip
5
+
6
  gpu_info = getoutput('nvidia-smi')
7
  if("A10G" in gpu_info):
8
+ os.system(f"pip install -q https://github.com/camenduru/stable-diffusion-webui-colab/releases/download/0.0.16/xformers-0.0.16+814314d.d20230119.A10G-cp310-cp310-linux_x86_64.whl")
9
  elif("T4" in gpu_info):
10
+ os.system(f"pip install -q https://github.com/camenduru/stable-diffusion-webui-colab/releases/download/0.0.16/xformers-0.0.16+814314d.d20230119.A10G-cp310-cp310-linux_x86_64.whl")
11
 
12
  os.system(f"git clone -b v1.5 https://github.com/camenduru/stable-diffusion-webui /home/user/app/stable-diffusion-webui")
13
  os.chdir("/home/user/app/stable-diffusion-webui")