diff --git a/nodes.py b/nodes.py
index eefa861..62f827b 100644
--- a/nodes.py
+++ b/nodes.py
@@ -71,11 +71,11 @@ class DownloadAndLoadCogVideoModel:
         pipe = CogVideoXPipeline(vae, transformer, scheduler)
 
         if torch_compile:
+            torch._dynamo.config.suppress_errors = True
             torch._inductor.config.conv_1x1_as_mm = True
             torch._inductor.config.coordinate_descent_tuning = True
             torch._inductor.config.epilogue_fusion = False
             torch._inductor.config.coordinate_descent_check_all_directions = True
-            torch._inductor.config.num_stages = 1
             pipe.transformer.to(memory_format=torch.channels_last)
             pipe.transformer = torch.compile(pipe.transformer, mode="max-autotune", fullgraph=True)