Omnibus commited on
Commit
a0e32a4
1 Parent(s): 94cc393

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -30,7 +30,7 @@ main_pipe = StableDiffusionControlNetPipeline.from_pretrained(
30
  vae=vae,
31
  safety_checker=None,
32
  torch_dtype=torch.float16,
33
- ).to("cuda")
34
  #main_pipe.unet = torch.compile(main_pipe.unet, mode="reduce-overhead", fullgraph=True)
35
  #main_pipe.unet.to(memory_format=torch.channels_last)
36
  #main_pipe.unet = torch.compile(main_pipe.unet, mode="reduce-overhead", fullgraph=True)
 
30
  vae=vae,
31
  safety_checker=None,
32
  torch_dtype=torch.float16,
33
+ ).to("cpu")
34
  #main_pipe.unet = torch.compile(main_pipe.unet, mode="reduce-overhead", fullgraph=True)
35
  #main_pipe.unet.to(memory_format=torch.channels_last)
36
  #main_pipe.unet = torch.compile(main_pipe.unet, mode="reduce-overhead", fullgraph=True)