AMD/ROCm - Fix VAE_KL_MEM_RATIO overestimation for modern ROCm - Added comment
This commit is contained in:
parent
2ec0fd0e00
commit
9aec99ed8a
|
|
@ -440,7 +440,7 @@ class VAE:
|
|||
sd = diffusers_convert.convert_vae_state_dict(sd)
|
||||
|
||||
if model_management.is_amd():
|
||||
VAE_KL_MEM_RATIO = 1.3
|
||||
VAE_KL_MEM_RATIO = 1.3 # Conservative margin for modern ROCm 7.x, reduced from 2.73 (see PR #12685)
|
||||
else:
|
||||
VAE_KL_MEM_RATIO = 1.0
|
||||
|
||||
|
|
|
|||
Loading…
Reference in New Issue