From acf99d639568ca85615b4945b2a8295d8d0973ec Mon Sep 17 00:00:00 2001
From: blepping <blepping@tuta.io>
Date: Fri, 22 Nov 2024 17:34:49 -0700
Subject: [PATCH] Fix an issue where bleh would not load without sageattention

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

diff --git a/py/nodes/sageAttention.py b/py/nodes/sageAttention.py
index 6b24708..753fa34 100644
--- a/py/nodes/sageAttention.py
+++ b/py/nodes/sageAttention.py
@@ -27,7 +27,7 @@ def attention_sage(  # noqa: PLR0917
     attn_precision=None,
     skip_reshape=False,
     sageattn_allow_head_sizes: None | set | tuple | list = None,
-    sageattn_function=sageattention.sageattn,
+    sageattn_function=sageattention.sageattn if sageattention is not None else None,
     sageattn_verbose=False,
     **kwargs: dict[str],
 ):