Hugging Face's logo

Spaces:
nightfury
/
Stable_Diffusion
Runtime error

App Files Community
nightfury commited on
Commit
66c43ab
·
1 Parent(s): 304b052

Update dj_sd.py

Browse files
Files changed (1) hide show
  1. dj_sd.py +1 -1
dj_sd.py CHANGED
@@ -13,7 +13,7 @@ import requests
13
  #from share_btn import community_icon_html, loading_icon_html, share_js
14
 
15
  model_id = "CompVis/stable-diffusion-v1-4"
16
- device = "cuda"
17
  MY_SECRET_TOKEN=os.environ.get('HF_TOKEN_SD')
18
  YOUR_TOKEN=MY_SECRET_TOKEN
19
  #If you are running this code locally, you need to either do a 'huggingface-cli login` or paste your User Access Token from here https://huggingface.co/settings/tokens into the use_auth_token field below.
 
13
  #from share_btn import community_icon_html, loading_icon_html, share_js
14
 
15
  model_id = "CompVis/stable-diffusion-v1-4"
16
+ device = "cpu"
17
  MY_SECRET_TOKEN=os.environ.get('HF_TOKEN_SD')
18
  YOUR_TOKEN=MY_SECRET_TOKEN
19
  #If you are running this code locally, you need to either do a 'huggingface-cli login` or paste your User Access Token from here https://huggingface.co/settings/tokens into the use_auth_token field below.