wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2025-12-22 20:40:49 +08:00
Code Issues Actions 7 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-20 00:10:20 +03:00
parent 852704c81a
commit 3988f37386
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
comfy/model_management.py
View File

@ -1112,7 +1112,7 @@ def pytorch_attention_enabled():
return ENABLE_PYTORCH_ATTENTION return ENABLE_PYTORCH_ATTENTION
def pytorch_attention_enabled_vae(): 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 False # enabling pytorch attention on AMD currently causes crash when doing high res
return pytorch_attention_enabled() return pytorch_attention_enabled()