From 58668a44cc4e179a59dec5684a6d0295f5fad8e7 Mon Sep 17 00:00:00 2001
From: kijai <40791699+kijai@users.noreply.github.com>
Date: Sat, 19 Apr 2025 16:55:05 +0300
Subject: [PATCH] Update model_optimization_nodes.py

---
 nodes/model_optimization_nodes.py | 1 -
 1 file changed, 1 deletion(-)

diff --git a/nodes/model_optimization_nodes.py b/nodes/model_optimization_nodes.py
index 52c7d77..54e01ac 100644
--- a/nodes/model_optimization_nodes.py
+++ b/nodes/model_optimization_nodes.py
@@ -53,7 +53,6 @@ class BaseLoaderKJ:
 
             @torch.compiler.disable()
             def attention_sage(q, k, v, heads, mask=None, attn_precision=None, skip_reshape=False, skip_output_reshape=False):
-                print("SAGE")
                 if skip_reshape:
                     b, _, _, dim_head = q.shape
                     tensor_layout="HND"