Merge ed4901d316 into 7ee77ff038

Merge branch 'master' into pytorch-attention-vae-rdna4
Enable pytorch attention in VAE for AMD RDNA 4
2026-02-06 19:42:34 +08:00 · 2026-01-26 14:29:31 +08:00 · 2026-01-23 18:28:58 +02: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	06dfdf8747	Merge `ed4901d316` into `7ee77ff038`	2026-01-26 14:29:31 +08:00
DELUXA	ed4901d316	Merge branch 'master' into pytorch-attention-vae-rdna4	2026-01-23 18:28:58 +02:00
Your Name	3988f37386	Enable pytorch attention in VAE for AMD RDNA 4	2025-09-20 00:10:20 +03:00