Update nodes.py

2024-08-28 00:56:15 +03:00
parent 4a75855740
commit 58fee8e567
1 changed files with 1 additions and 1 deletions
@@ -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)