Patch Sage Attention KJ Node Error - After ComfyUI version: 0.3.59 Update

[monnky]

If we use Patch Sage Attention KJ Node the following error appears.

TypeError: BaseLoaderKJ._patch_modules..attention_sage() got an unexpected keyword argument 'transformer_options'

Presently, disabling the node only solution, at the cost of speed.

[agusliem]

yeah i got the same problem... But i add --use-sage-attention command for run.bat comfyui

.\python_embeded\python.exe -s ComfyUI\main.py --windows-standalone-build --use-sage-attention
pause

[kijai]

It was fixed already couple of days ago, you just need to update these nodes to nightly version.

[agusliem]

It was fixed already couple of days ago, you just need to update these nodes to nightly version.

Awesome Thank you so much Kijai.

[monnky]

thank you , king kijai. youre a lifesaver.

[franckdsf]

@kijai hey, It's not fixed for me. Updating to 1.1.5 gives me the following error:

[ERROR] An error occurred while retrieving information for the 'GGUFLoaderKJ' node.
Traceback (most recent call last):
  File "/workspace/comfyui/custom_nodes/ComfyUI-KJNodes/nodes/model_optimization_nodes.py", line 1929, in _ensure_gguf_loaded
    cls.gguf_nodes = importlib.import_module("ComfyUI-GGUF")
                     ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/opt/conda/lib/python3.11/importlib/__init__.py", line 126, in import_module
    return _bootstrap._gcd_import(name[level:], package, level)
           ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "<frozen importlib._bootstrap>", line 1204, in _gcd_import
  File "<frozen importlib._bootstrap>", line 1176, in _find_and_load
  File "<frozen importlib._bootstrap>", line 1140, in _find_and_load_unlocked
ModuleNotFoundError: No module named 'ComfyUI-GGUF'

During handling of the above exception, another exception occurred:

Traceback (most recent call last):
  File "/workspace/comfyui/custom_nodes/ComfyUI-KJNodes/nodes/model_optimization_nodes.py", line 1932, in _ensure_gguf_loaded
    cls.gguf_nodes = importlib.import_module("comfyui-gguf")
                     ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/opt/conda/lib/python3.11/importlib/__init__.py", line 126, in import_module
    return _bootstrap._gcd_import(name[level:], package, level)
           ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "<frozen importlib._bootstrap>", line 1204, in _gcd_import
  File "<frozen importlib._bootstrap>", line 1176, in _find_and_load
  File "<frozen importlib._bootstrap>", line 1140, in _find_and_load_unlocked
ModuleNotFoundError: No module named 'comfyui-gguf'

During handling of the above exception, another exception occurred:

Traceback (most recent call last):
  File "/workspace/comfyui/server.py", line 629, in get_object_info
    out[x] = node_info(x)
             ^^^^^^^^^^^^
  File "/workspace/comfyui/server.py", line 591, in node_info
    return obj_class.GET_NODE_INFO_V1()
           ^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/workspace/comfyui/comfy_api/latest/_io.py", line 1315, in GET_NODE_INFO_V1
    schema = cls.GET_SCHEMA()
             ^^^^^^^^^^^^^^^^
  File "/workspace/comfyui/comfy_api/latest/_io.py", line 1440, in GET_SCHEMA
    schema = cls.FINALIZE_SCHEMA()
             ^^^^^^^^^^^^^^^^^^^^^
  File "/workspace/comfyui/comfy_api/latest/_io.py", line 1431, in FINALIZE_SCHEMA
    schema = cls.define_schema()
             ^^^^^^^^^^^^^^^^^^^
  File "/workspace/comfyui/custom_nodes/ComfyUI-KJNodes/nodes/model_optimization_nodes.py", line 1938, in define_schema
    cls._ensure_gguf_loaded()
  File "/workspace/comfyui/custom_nodes/ComfyUI-KJNodes/nodes/model_optimization_nodes.py", line 1934, in _ensure_gguf_loaded
    raise ImportError("This node requires ComfyUI-GGUF to be installed.")
ImportError: This node requires ComfyUI-GGUF to be installed.

But I have the ComfyUI-GGUF node (v1.1.4). I'm loading my model with the Unet GGUF Loader (UnetLoaderGGUF)

I even have ComfyUI-GGUF prompting : ComfyUI-GGUF: Allowing full torch compile at the start of Comfy

Edit: It seems to be working, this is a just a warning about your new node

[LIQUIDMIND111]

I changed my GPU from RTX 3060 to 5060 ti and now i have this TORCH INDUCTOR ERROR that won't go away

i had to REVERT on COMFY, and no issues with sage ATTENTION at all, the ONLY issue is the TORCH NODE, inductor.... I even FIXED the COMFY update issue by updating SAGE ATTENTION, and thats fine, but now using the NEW comfy update or even reverting to sept 4 branch, still gives me TORCH INDUCTOR ERROR,

so its something related to the 5060 ti gpu!!

[123LiVo321]

as of today - 25.92025 I encountered the same error - TypeError: BaseLoaderKJ._patch_modules..attention_sage() got an unexpected keyword argument 'transformer_options'.
I have updated comfyui, updated frontend, updated efficiency nodes to nightly version, sage attention used with parameter '--use-sage-attention' (not the patch node)

[kijai]

as of today - 25.92025 I encountered the same error - TypeError: BaseLoaderKJ._patch_modules..attention_sage() got an unexpected keyword argument 'transformer_options'.
I have updated comfyui, updated frontend, updated efficiency nodes to nightly version, sage attention used with parameter '--use-sage-attention' (not the patch node)

That error is from the old version of the patch node, it was fixed 2 weeks ago in these nodes.

[mib5ht]

updating kjNOdes fixed

[KLL535]

Same error.
I have global flags "--use-sage-attention" and "--fast" for all ksamplers.

[LaloHao]

I changed my GPU from RTX 3060 to 5060 ti and now i have this TORCH INDUCTOR ERROR that won't go away

i had to REVERT on COMFY, and no issues with sage ATTENTION at all, the ONLY issue is the TORCH NODE, inductor.... I even FIXED the COMFY update issue by updating SAGE ATTENTION, and thats fine, but now using the NEW comfy update or even reverting to sept 4 branch, still gives me TORCH INDUCTOR ERROR,

so its something related to the 5060 ti gpu!!

You need to update torch* to use a 50xx GPU

pip install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu129