Use larger atol in `torch.allclose` for some tests (#21966)

Use larger atol Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>

Use larger atol in `torch.allclose` for some tests (#21966)
Use larger atol Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
9474abdf · Yih-Dar · GitHub · 64d95c44 · 9474abdf
Unverified Commit 9474abdf authored Mar 06, 2023 by Yih-Dar Committed by GitHub Mar 06, 2023
Show whitespace changes
Inline Side-by-side

Showing with 3 additions and 3 deletions

tests/test_modeling_common.py tests/test_modeling_common.py +3 -3

No files found.
--- a/tests/test_modeling_common.py
+++ b/tests/test_modeling_common.py
@@ -2497,7 +2497,7 @@ class ModelTesterMixin:
                torch.manual_seed(0)
                new_output = new_model(**inputs_dict_class)

-                self.assertTrue(torch.allclose(base_output[0], new_output[0]))
+                self.assertTrue(torch.allclose(base_output[0], new_output[0], atol=1e-5))

    @require_accelerate
    @mark.accelerate_tests
@@ -2533,7 +2533,7 @@ class ModelTesterMixin:
                    torch.manual_seed(0)
                    new_output = new_model(**inputs_dict_class)

-                    self.assertTrue(torch.allclose(base_output[0], new_output[0]))
+                    self.assertTrue(torch.allclose(base_output[0], new_output[0], atol=1e-5))

    @require_accelerate
    @mark.accelerate_tests
@@ -2569,7 +2569,7 @@ class ModelTesterMixin:
                    torch.manual_seed(0)
                    new_output = new_model(**inputs_dict_class)

-                    self.assertTrue(torch.allclose(base_output[0], new_output[0]))
+                    self.assertTrue(torch.allclose(base_output[0], new_output[0], atol=1e-5))

    def test_problem_types(self):
        config, inputs_dict = self.model_tester.prepare_config_and_inputs_for_common()