nightfury commited on
Commit
1730273
·
1 Parent(s): 8549305

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -14,7 +14,7 @@ print("hello !")
14
 
15
  YOUR_TOKEN=MY_SECRET_TOKEN
16
 
17
- device="cuda"
18
 
19
  #prompt_pipe = StableDiffusionPipeline.from_pretrained("CompVis/stable-diffusion-v1-4", use_auth_token=YOUR_TOKEN)
20
  #prompt_pipe.to(device)
 
14
 
15
  YOUR_TOKEN=MY_SECRET_TOKEN
16
 
17
+ device="cpu"
18
 
19
  #prompt_pipe = StableDiffusionPipeline.from_pretrained("CompVis/stable-diffusion-v1-4", use_auth_token=YOUR_TOKEN)
20
  #prompt_pipe.to(device)