diff --git a/comfy/model_management.py b/comfy/model_management.py
index 232d363aa..1f6445c9f 100644
--- a/comfy/model_management.py
+++ b/comfy/model_management.py
@@ -1044,6 +1044,9 @@ def cast_to_device(tensor, device, dtype, copy=False):
 def sage_attention_enabled():
     return args.use_sage_attention
 
+def sage_attention3_enabled():
+    return args.use_sage_attention3
+
 def flash_attention_enabled():
     return args.use_flash_attention