Update expected values in MgpstrModelIntegrationTest (#22195)

Update values Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
2023-03-16 12:48:52 +01:00
parent 1485bd9c02
commit 52a57f7c7c
1 changed files with 1 additions and 1 deletions
--- a/tests/models/mgp_str/test_modeling_mgp_str.py
+++ b/tests/models/mgp_str/test_modeling_mgp_str.py
@@ -262,7 +262,7 @@ class MgpstrModelIntegrationTest(unittest.TestCase):
        self.assertEqual(out_strs["generated_text"][0], expected_text)

        expected_slice = torch.tensor(
-            [[[-39.7358, -44.8562, -36.6253], [-62.3605, -64.5908, -59.0069], [-74.6127, -68.9724, -71.7150]]],
+            [[[-39.5397, -44.4024, -36.1844], [-61.4709, -63.8639, -58.3454], [-74.0225, -68.5494, -71.2164]]],
            device=torch_device,
        )