AMD/ROCm - Fix VAE_KL_MEM_RATIO overestimation for modern ROCm #2

This commit is contained in:
ReinerBforartists 2026-02-27 20:11:26 +01:00
parent 1f1ec377ce
commit 2ec0fd0e00

View File

@ -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