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"