.

app.py

@@ -60,7 +60,7 @@ pipe.to(device=DEVICE)
 def compile_em():
     pipe.unet = torch.compile(pipe.unet, mode='reduce-overhead')
     pipe.vae = torch.compile(pipe.vae, mode='reduce-overhead')
-    autoencoder.model.forward = torch.compile(autoencoder.model.forward, backend='inductor', dynamic=True, mode='reduce-overhead')
+    autoencoder.model.forward = torch.compile(autoencoder.model.forward, backend='inductor', dynamic=True)
 
 
 output_hidden_state = False