Jccqqqqq commited on
Commit
9a53bc3
·
1 Parent(s): d6bc7c8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -16,6 +16,7 @@ pipe = StableDiffusionControlNetPipeline.from_pretrained(
16
  controlnet=controlnet,
17
  torch_dtype=torch.float16,
18
  )
 
19
  pipe.scheduler = UniPCMultistepScheduler.from_config(pipe.scheduler.config)
20
  pipe.enable_model_cpu_offload()
21
  generators = []
 
16
  controlnet=controlnet,
17
  torch_dtype=torch.float16,
18
  )
19
+ pipe.to("cpu")
20
  pipe.scheduler = UniPCMultistepScheduler.from_config(pipe.scheduler.config)
21
  pipe.enable_model_cpu_offload()
22
  generators = []