diff --git a/tests/models/test_models_vae.py b/tests/models/test_models_vae.py
index 95e5f656af..f6333d6cd9 100644
--- a/tests/models/test_models_vae.py
+++ b/tests/models/test_models_vae.py
@@ -209,7 +209,7 @@ class AutoencoderKLIntegrationTests(unittest.TestCase):
         output_slice = sample[-1, -2:, :2, -2:].flatten().float().cpu()
         expected_output_slice = torch.tensor(expected_slice)
 
-        assert torch_all_close(output_slice, expected_output_slice, atol=5e-3)
+        assert torch_all_close(output_slice, expected_output_slice, atol=1e-2)
 
     @parameterized.expand(
         [