Bradarr commited on
Commit
ed44d2c
·
1 Parent(s): 9cb0363

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -3,7 +3,9 @@ import requests
3
  import json
4
  import os
5
 
6
- os.system(f"pip install torch torchvision pprint")
 
 
7
  os.system(f"pip install git+https://github.com/huggingface/transformers")
8
  #os.system(f"git clone https://github.com/camenduru/stable-diffusion-webui /home/user/app/stable-diffusion-webui")
9
 
 
3
  import json
4
  import os
5
 
6
+
7
+ #os.system(f"pip install torch torchvision")
8
+ os.system(f"pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu116")
9
  os.system(f"pip install git+https://github.com/huggingface/transformers")
10
  #os.system(f"git clone https://github.com/camenduru/stable-diffusion-webui /home/user/app/stable-diffusion-webui")
11