lsb commited on
Commit
0025e75
1 Parent(s): 4380df2

no compiling for now

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -35,9 +35,9 @@ inpainting_pipeline = StableDiffusionInpaintPipeline.from_pretrained(
35
  safety_checker=None,
36
  ).to(inpaint_preferred_device)
37
 
38
- inpainting_pipeline.unet = torch.compile(inpainting_pipeline.unet, backend=preferred_backend)
39
- inpainting_pipeline.vae = torch.compile(inpainting_pipeline.vae, backend=preferred_backend)
40
- seg_model = torch.compile(seg_model, backend=preferred_backend)
41
 
42
  seg_working_size = (seg_model_img_size, seg_model_img_size)
43
  repaint_working_size = (768, 768)
 
35
  safety_checker=None,
36
  ).to(inpaint_preferred_device)
37
 
38
+ #inpainting_pipeline.unet = torch.compile(inpainting_pipeline.unet, backend=preferred_backend)
39
+ #inpainting_pipeline.vae = torch.compile(inpainting_pipeline.vae, backend=preferred_backend)
40
+ #seg_model = torch.compile(seg_model, backend=preferred_backend)
41
 
42
  seg_working_size = (seg_model_img_size, seg_model_img_size)
43
  repaint_working_size = (768, 768)