From 6ce01bd647c90edb520aaf2c2aeaf9be144bee75 Mon Sep 17 00:00:00 2001 From: Dhruv Nair Date: Fri, 6 Oct 2023 10:25:18 +0200 Subject: [PATCH] Bump tolerance on shape test (#5289) bump tolerance on shape test --- tests/pipelines/shap_e/test_shap_e_img2img.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/pipelines/shap_e/test_shap_e_img2img.py b/tests/pipelines/shap_e/test_shap_e_img2img.py index 55c0ae6bd0..055dbe7a97 100644 --- a/tests/pipelines/shap_e/test_shap_e_img2img.py +++ b/tests/pipelines/shap_e/test_shap_e_img2img.py @@ -250,7 +250,7 @@ class ShapEImg2ImgPipelineFastTests(PipelineTesterMixin, unittest.TestCase): super().test_float16_inference(expected_max_diff=1e-1) def test_save_load_local(self): - super().test_save_load_local(expected_max_difference=1e-3) + super().test_save_load_local(expected_max_difference=5e-3) @unittest.skip("Key error is raised with accelerate") def test_sequential_cpu_offload_forward_pass(self):