mirror of
https://github.com/comfyanonymous/ComfyUI.git
synced 2026-03-16 22:58:19 +08:00
AMD/ROCm - Fix VAE_KL_MEM_RATIO overestimation for modern ROCm #2
This commit is contained in:
parent
1f1ec377ce
commit
2ec0fd0e00
@ -440,7 +440,7 @@ class VAE:
|
||||
sd = diffusers_convert.convert_vae_state_dict(sd)
|
||||
|
||||
if model_management.is_amd():
|
||||
VAE_KL_MEM_RATIO = 2.73
|
||||
VAE_KL_MEM_RATIO = 1.3
|
||||
else:
|
||||
VAE_KL_MEM_RATIO = 1.0
|
||||
|
||||
|
||||
Loading…
Reference in New Issue
Block a user