From 7364ce7d2d91a9bd7b4a44757c78191cdfecdfcb Mon Sep 17 00:00:00 2001
From: kijai <40791699+kijai@users.noreply.github.com>
Date: Sun, 12 Jan 2025 19:42:19 +0200
Subject: [PATCH] remove prints

---
 nodes/model_optimization_nodes.py | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/nodes/model_optimization_nodes.py b/nodes/model_optimization_nodes.py
index 5213c27..43fb7de 100644
--- a/nodes/model_optimization_nodes.py
+++ b/nodes/model_optimization_nodes.py
@@ -490,7 +490,7 @@ class TorchCompileCosmosModel:
         if not self._compiled:
             try:
                 for name, block in diffusion_model.blocks.items():
-                    print(f"Compiling block {name}")
+                    #print(f"Compiling block {name}")
                     compiled_block = torch.compile(block, mode=mode, dynamic=dynamic, fullgraph=fullgraph, backend=backend)
                     m.add_object_patch(f"diffusion_model.blocks.{name}", compiled_block)
                     #diffusion_model.blocks[name] = compiled_block
@@ -503,7 +503,6 @@ class TorchCompileCosmosModel:
                     "dynamic": dynamic,
                 }
                 setattr(m.model, "compile_settings", compile_settings)
-                print(model.model.diffusion_model.blocks)
                
             except:
                 raise RuntimeError("Failed to compile model")