no compiling for now
Browse files
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)
|