Update src/pipeline.py

src/pipeline.py

@@ -12,6 +12,7 @@ def load_pipeline() -> StableDiffusionXLPipeline:
         torch_dtype=torch.float16,
     )
     pipe.vae = AutoencoderTiny.from_pretrained("madebyollin/taesdxl", torch_dtype=torch.float16)
+    pipeline.unet = torch.compile(pipeline.unet, mode='reduce-overhead', fullgraph=True)
     pipe.to("cuda")
    
     return pipeline