Nodes/ComfyUI-DistorchMemoryManager/Patch Sage Attention DM
ComfyUI Node

Patch Sage Attention DM

Experimental node for patching attention mode. This doesn't use the model patching system and thus can't be disabled without running the node again with 'disabled' option.

By ussoewwin·Created 10 months ago·Updated 27 days ago· 42
Patch Sage Attention DM
  • model
  • MODEL
sage_attentionfalse
allow_compilefalse
CategoryMemory

Inputs (3)

NameTypeDefaultDescription
modelMODEL
sage_attentionCOMBOfalseGlobal patch comfy attention to use sageattn, once patched to revert back to normal you would need to run this node again with disabled option.
allow_compileoptBOOLEANfalseAllow the use of torch.compile for the sage attention function, requires latest sageattn 2.2.0 or higher.

Outputs (1)

NameTypeDescription
MODELMODEL