From 58fee8e567e08066a04df9a6e4a3f5eea7d6d071 Mon Sep 17 00:00:00 2001
From: kijai <40791699+kijai@users.noreply.github.com>
Date: Wed, 28 Aug 2024 00:56:15 +0300
Subject: [PATCH] Update nodes.py

---
 nodes.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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)