之前安裝SageAttention後，一直不能使用，在vae的地方出現了Unsupported head_dim: 384的問題，我還特地問SageAttention windows whl的作者
感謝這作者製作windows版的sageAttention跟triton
對方給出的回應是說SageAttention只支持head_dim <= 128的模型，因此不適用所有地方。所以我原本是認為我那個工作流不能用SageAttention才報錯的。
但後來最近使用了wan2.2的某個工作流，這個工作流確定是要使用SageAttention來加速的，但一樣出現了Unsupported head_dim: 384的問題。
到這個節點的github查詢後，發現有人在問同一件事情
根據這個節點的作者kijai大神所說，這跟這節點沒關係，應該是有某自訂節點影響了，讓全局都使用了SageAttention，在那篇文章有人是說Hi3dGen 這個節點影響的，但我並沒有裝這個節點，因此寫了一個小腳本來判斷，看看哪些可能有問題
import os
import re
root = "F:\project\AI\ComfyUI\ComfyUI_windows_portable_nvidia\portable_251003\ComfyUI\custom_nodes" #替換成自己的路徑
patterns = [
r"scaled_dot_product_attention",
r"sageattention",
r"F\.scaled_dot_product_attention",
r"import sageattention"
]
print("???? 掃描 custom_nodes 目錄中有修改 attention 的節點...\n")
for dirpath, _, filenames in os.walk(root):
for f in filenames:
#print (filenames)
if f.endswith(".py"):
path = os.path.join(dirpath, f)
try:
text = open(path, "r", encoding="utf-8").read()
except:
continue
for p in patterns:
if re.search(p, text):
print(f"⚠️ 發現可疑檔案：{path}")
break
print("\n✅ 掃描完成。若有顯示 ⚠️，表示該節點可能覆蓋 attention。")我將檢查結果丟給chatGPT分析，判斷出了以下幾個可能會影響的節點
果然，當我移除掉ComfyUI-IF_Trellis這個自訂節點後，SageAttention就可以使用了，圖片中的其他節點可以繼續保留著。
題外話，當初問了chatGPT、gemini、grok 都跟我說只能關閉掉SageAttention，要不是有這次的引導，它根本不給我正確解答。
文章轉載或引用，請先告知並保留原文出處與連結！！(單純分享或非營利的只需保留原文出處，不用告知)
原文連結:
https://blog.aidec.tw/post/sageattention-comfyui-problem
若有業務合作需求，可寫信至： [email protected]
創業、網站經營相關內容未來將發布在 小易創業筆記