Furkan12 commited on
Commit
c6cb106
1 Parent(s): e88405d
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -17,7 +17,7 @@ model_id = "stabilityai/stable-diffusion-2-1"
17
  # Use the DPMSolverMultistepScheduler (DPM-Solver++) scheduler here instead
18
  pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
19
  pipe.scheduler = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config)
20
- pipe = pipe.to("cuda")
21
 
22
  def resize(value,img):
23
  img = Image.open(img)
 
17
  # Use the DPMSolverMultistepScheduler (DPM-Solver++) scheduler here instead
18
  pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
19
  pipe.scheduler = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config)
20
+ pipe = pipe.to("cpu")
21
 
22
  def resize(value,img):
23
  img = Image.open(img)