torch compile

app.py

@@ -30,6 +30,9 @@ inpainting_pipeline = StableDiffusionInpaintPipeline.from_pretrained(
     safety_checker=None,
 ).to(preferred_device)
 
+inpainting_pipeline.unet = torch.compile(inpainting_pipeline.unet)
+inpainting_pipeline.vae = torch.compile(inpainting_pipeline.vae)
+
 # inpainting_pipeline.scheduler = LCMScheduler.from_config(inpainting_pipeline.scheduler.config)
 # inpainting_pipeline.load_lora_weights("latent-consistency/lcm-lora-sdv1-5", torch_dtype=preferred_dtype)
 # inpainting_pipeline.fuse_lora()