Merge 3988f37386 into 2c03884f5f

Enable pytorch attention in VAE for AMD RDNA 4
2026-01-24 13:20:19 +08:00 · 2026-01-06 16:55:47 -07: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
@ -1241,7 +1241,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	b2bb7b2183	Merge `3988f37386` into `2c03884f5f`	2026-01-06 16:55:47 -07:00
Your Name	3988f37386	Enable pytorch attention in VAE for AMD RDNA 4	2025-09-20 00:10:20 +03:00