Merge 3988f37386 into 4484b93d61

Enable pytorch attention in VAE for AMD RDNA 4
2026-01-28 15:20:25 +08:00 · 2026-01-09 14:46:06 -06:00 · 2025-09-20 00:10:20 +03:00
1 changed files with 1 additions and 1 deletions
--- a/comfy/model_management.py
+++ b/comfy/model_management.py
@ -1252,7 +1252,7 @@ def pytorch_attention_enabled():
    return ENABLE_PYTORCH_ATTENTION

 def pytorch_attention_enabled_vae():
-    if is_amd():
+    if is_amd() and not amd_min_version(device=None, min_rdna_version=4):
        return False  # enabling pytorch attention on AMD currently causes crash when doing high res
    return pytorch_attention_enabled()
Author	SHA1	Message	Date
DELUXA	1eb39fe852	Merge `3988f37386` into `4484b93d61`	2026-01-09 14:46:06 -06:00
Your Name	3988f37386	Enable pytorch attention in VAE for AMD RDNA 4	2025-09-20 00:10:20 +03:00