Bartusito commited on
Commit
9dc356c
·
1 Parent(s): 0c02385

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -19,7 +19,7 @@ def modelo2(text):
19
  # Use the DPMSolverMultistepScheduler (DPM-Solver++) scheduler here instead
20
  pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
21
  pipe.scheduler = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config)
22
- pipe = pipe.to("cuda")
23
 
24
  image = pipe(text).images[0]
25
  return image
 
19
  # Use the DPMSolverMultistepScheduler (DPM-Solver++) scheduler here instead
20
  pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
21
  pipe.scheduler = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config)
22
+ pipe = pipe.to("cpu")
23
 
24
  image = pipe(text).images[0]
25
  return image