Update expected values in MgpstrModelIntegrationTest (huggingface#2…

…2195) Update values Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
novice03 · Jun 23, 2023 · 383b3fe · 383b3fe
1 parent c2ef58f
commit 383b3fe
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/models/mgp_str/test_modeling_mgp_str.py b/tests/models/mgp_str/test_modeling_mgp_str.py
@@ -262,7 +262,7 @@ def test_inference(self):
         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,
         )