Merge 3988f37386 into abe2ec26a6

Enable pytorch attention in VAE for AMD RDNA 4
2026-02-06 11:32:31 +08:00 · 2026-01-21 18:48:07 -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	fd683c4cf7	Merge `3988f37386` into `abe2ec26a6`	2026-01-21 18:48:07 -06:00
Your Name	3988f37386	Enable pytorch attention in VAE for AMD RDNA 4	2025-09-20 00:10:20 +03:00