xinyun/ComfyUI
1
0
Fork 0
mirror of https://git.datalinker.icu/comfyanonymous/ComfyUI synced 2026-08-14 07:50:04 +08:00
Code Issues Packages Projects Releases Wiki Activity

Enable pytorch attention in VAE for AMD RDNA 4

Browse Source
This commit is contained in:
Your Name 2025-09-19 23:51:58 +03:00
parent dd611a7700
commit ba50d368c3
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
comfy/model_management.py
View File

@ -348,7 +348,7 @@ try:
# if any((a in arch) for a in ["gfx1201"]):
# ENABLE_PYTORCH_ATTENTION = True
if torch_version_numeric >= (2, 7) and rocm_version >= (6, 4):
if any((a in arch) for a in ["gfx1201", "gfx942", "gfx950"]): # TODO: more arches
if any((a in arch) for a in ["gfx1200", "gfx1201", "gfx942", "gfx950"]): # TODO: more arches
SUPPORT_FP8_OPS = True
except:
@ -1112,7 +1112,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()