Hugging Face's logo

Spaces:
vsanimator
/
sketch-a-sketch-demo
Runtime error

App Files Community
1
Vishnu Sarukkai commited on
Commit
17cb476
•
1 Parent(s): d97135c

Gpu

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -7,7 +7,7 @@ from PIL import Image
7
  import os
8
 
9
  negative_prompt = ""
10
- device = 'cpu'
11
  controlnet = ControlNetModel.from_pretrained("vsanimator/sketch-a-sketch").to(device)
12
  pipe = StableDiffusionControlNetPipeline.from_pretrained(
13
  "runwayml/stable-diffusion-v1-5",
 
7
  import os
8
 
9
  negative_prompt = ""
10
+ device = 'cuda:0'
11
  controlnet = ControlNetModel.from_pretrained("vsanimator/sketch-a-sketch").to(device)
12
  pipe = StableDiffusionControlNetPipeline.from_pretrained(
13
  "runwayml/stable-diffusion-v1-5",