From 5e1865c1c7cc75601b6c4cfbb2ebd717df306f9a Mon Sep 17 00:00:00 2001 From: Panchovix Date: Sat, 26 Jul 2025 14:21:48 -0400 Subject: [PATCH] Separate sage 1.x/2.x and sage 3.x on different functions and flags 2. --- comfy/cli_args.py | 1 + 1 file changed, 1 insertion(+) diff --git a/comfy/cli_args.py b/comfy/cli_args.py index 0d760d524..e511cee60 100644 --- a/comfy/cli_args.py +++ b/comfy/cli_args.py @@ -111,6 +111,7 @@ attn_group.add_argument("--use-split-cross-attention", action="store_true", help attn_group.add_argument("--use-quad-cross-attention", action="store_true", help="Use the sub-quadratic cross attention optimization . Ignored when xformers is used.") attn_group.add_argument("--use-pytorch-cross-attention", action="store_true", help="Use the new pytorch 2.0 cross attention function.") attn_group.add_argument("--use-sage-attention", action="store_true", help="Use sage attention.") +attn_group.add_argument("--use-sage-attention3", action="store_true", help="Use sage attention 3. Supported only on blackwell GPUs.") attn_group.add_argument("--use-flash-attention", action="store_true", help="Use FlashAttention.") parser.add_argument("--disable-xformers", action="store_true", help="Disable xformers.")