Extensions/ComfyUI-SageAttention3
ComfyUI Extension

ComfyUI-SageAttention3

Strict Sage 3 (Blackwell) attention for ComfyUI official flows.

By wallen0322·Created 10 months ago·Updated 7 months ago· 26
wallen0322/ComfyUI-SageAttention3
Nodes1
On cloudLocal install
Categoryattention/Sage3
Stars26
Updated7 months ago

Nodes (1)

Attention: Sage 3 (Blackwell)
attention/Sage3
Readme

ComfyUI-SageAttention3

An experimental node

Strict Sage 3 (Blackwell) attention for ComfyUI official flows.

  • Hot-patches torch.nn.functional.scaled_dot_product_attention to sageattn3.api.sageattn3_blackwell
  • Prints which backend is active (toggleable)

Install

  1. Copy ComfyUI-SageAttention3 into ComfyUI/custom_nodes/
  2. Install: pip install sageattn3
  3. install from repo - https://github.com/thu-ml/SageAttention/tree/main/sageattention3_blackwell

Use

Add Attention: Sage 3 (Blackwell) before samplers, pass MODEL through, set enable=True.