Hugging Face's logo

Spaces:
Vsevolod
/
hack_spb
Runtime error

App Files Community
Vsevolod commited on
Commit
eed0fbd
•
1 Parent(s): 3b4e48b

cpu

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -6,7 +6,7 @@ from PIL import Image
6
  from diffusers import StableDiffusionImg2ImgPipeline
7
 
8
  # load the pipeline
9
- device = "cuda"
10
  model_id_or_path = "CompVis/stable-diffusion-v1-4"
11
  pipe = StableDiffusionImg2ImgPipeline.from_pretrained(
12
  model_id_or_path,
 
6
  from diffusers import StableDiffusionImg2ImgPipeline
7
 
8
  # load the pipeline
9
+ device = "cpu"#"cuda"
10
  model_id_or_path = "CompVis/stable-diffusion-v1-4"
11
  pipe = StableDiffusionImg2ImgPipeline.from_pretrained(
12
  model_id_or_path,