xinyun/ComfyUI-KJNodes
1
0
Fork 0
mirror of https://git.datalinker.icu/kijai/ComfyUI-KJNodes.git synced 2025-12-16 00:04:35 +08:00
Code Issues Packages Projects Releases Wiki Activity

Update model_optimization_nodes.py

Browse Source
This commit is contained in:
kijai 2025-01-16 21:13:37 +02:00
parent 5b8d419c6f
commit c9c8dcd5e7
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
nodes/model_optimization_nodes.py
View File

@ -125,12 +125,12 @@ class PathchSageAttentionKJ(BaseLoaderKJ):
def INPUT_TYPES(s): def INPUT_TYPES(s):
return {"required": { return {"required": {
"model": ("MODEL",), "model": ("MODEL",),
"sage_attention": (["disabled", "auto", "sageattn_qk_int8_pv_fp16_cuda", "sageattn_qk_int8_pv_fp16_triton", "sageattn_qk_int8_pv_fp8_cuda"], {"default": False, "tooltip": "Patch comfy attention to use sageattn."}), "sage_attention": (["disabled", "auto", "sageattn_qk_int8_pv_fp16_cuda", "sageattn_qk_int8_pv_fp16_triton", "sageattn_qk_int8_pv_fp8_cuda"], {"default": False, "tooltip": "Global patch comfy attention to use sageattn, once patched to revert back to normal you would need to run this node again with disabled option."}),
}} }}
RETURN_TYPES = ("MODEL", ) RETURN_TYPES = ("MODEL", )
FUNCTION = "patch" FUNCTION = "patch"
DESCRIPTION = "Experimental node for patching attention mode." DESCRIPTION = "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."
EXPERIMENTAL = True EXPERIMENTAL = True
CATEGORY = "KJNodes/experimental" CATEGORY = "KJNodes/experimental"