Fix an issue where bleh would not load without sageattention

2024-11-22 17:34:49 -07:00
parent 2ccc45b5b9
commit acf99d6395
1 changed files with 1 additions and 1 deletions
@@ -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],
 ):