COmpile is no longer the choice for this model

src/pipeline.py

@@ -52,7 +52,7 @@ def load_pipeline() -> Pipeline:
     pipeline.transformer.to(memory_format=torch.channels_last)
     # pipeline.vae.to(memory_format=torch.channels_last)
     quantize_(pipeline.vae, int8_weight_only())
-    pipeline.vae = torch.compile(pipeline.vae, mode="max-autotune")
+    # pipeline.vae = torch.compile(pipeline.vae, mode="max-autotune")
     pipeline.to("cuda")
     
     for _ in range(2):