From 41424466e36163e4f602cf54075166b861910c6f Mon Sep 17 00:00:00 2001 From: Dhruv Nair Date: Thu, 14 Mar 2024 10:24:01 +0530 Subject: [PATCH] [Tests] Fix incorrect constant in VAE scaling test. (#7301) update --- tests/models/autoencoders/test_models_vae.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/models/autoencoders/test_models_vae.py b/tests/models/autoencoders/test_models_vae.py index 8ddf6ac464..5321d8cc38 100644 --- a/tests/models/autoencoders/test_models_vae.py +++ b/tests/models/autoencoders/test_models_vae.py @@ -829,7 +829,7 @@ class AutoencoderKLIntegrationTests(unittest.TestCase): "https://huggingface.co/stabilityai/sd-vae-ft-mse-original/blob/main/vae-ft-mse-840000-ema-pruned.safetensors", ) - assert vae_default.config.scaling_factor == 0.18125 + assert vae_default.config.scaling_factor == 0.18215 assert vae_default.config.sample_size == 512 assert vae_default.dtype == torch.float32