From 1076d587b580bb0379037339fbb2bc796062192a Mon Sep 17 00:00:00 2001 From: Matt Date: Fri, 4 Nov 2022 12:45:34 +0000 Subject: [PATCH] Fix ESM LM head test (#20045) * Fix esm lm head test * make fixup --- tests/models/esm/test_modeling_esm.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/models/esm/test_modeling_esm.py b/tests/models/esm/test_modeling_esm.py index 1b6e024958..f6c0fcafb4 100644 --- a/tests/models/esm/test_modeling_esm.py +++ b/tests/models/esm/test_modeling_esm.py @@ -285,7 +285,7 @@ class EsmModelIntegrationTest(TestCasePlus): self.assertEqual(output.shape, expected_shape) expected_slice = torch.tensor( - [[[15.0973, -6.6406, -1.1351], [-0.2209, -9.9622, 4.2109], [-1.6055, -10.0023, 1.5914]]] + [[[8.9215, -10.5898, -6.4671], [-6.3967, -13.9114, -1.1212], [-7.7812, -13.9516, -3.7406]]] ) self.assertTrue(torch.allclose(output[:, :3, :3], expected_slice, atol=1e-4))