wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-04-20 15:32:32 +08:00
Code Issues Actions 14 Packages Projects Releases Wiki Activity

Merge 693ce0a1b2 into 138571da95

Browse Source
This commit is contained in:
apophis 2026-04-19 17:50:10 +08:00 committed by GitHub
commit 7918c09623
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
comfy/model_management.py
View File

@ -1446,7 +1446,7 @@ def pytorch_attention_enabled():
return ENABLE_PYTORCH_ATTENTION
def pytorch_attention_enabled_vae():
if is_amd():
if is_amd() and not SUPPORT_FP8_OPS: # exclude RDNA4 (gfx1200, gfx1201) and CDNA4 (gfx950) that support fp8
return False # enabling pytorch attention on AMD currently causes crash when doing high res
return pytorch_attention_enabled()

2
comfy/sd.py
View File

@ -440,7 +440,7 @@ class VAE:
if 'decoder.up_blocks.0.resnets.0.norm1.weight' in sd.keys(): #diffusers format
sd = diffusers_convert.convert_vae_state_dict(sd)
if model_management.is_amd():
if model_management.is_amd() and not model_management.SUPPORT_FP8_OPS: # exclude RDNA4 (gfx1200, gfx1201) and CDNA4 (gfx950) that support fp8
VAE_KL_MEM_RATIO = 2.73
else:
VAE_KL_MEM_RATIO = 1.0