From a80234c720465f5c06469656526f79b1daf33909 Mon Sep 17 00:00:00 2001 From: ydshieh Date: Thu, 16 Mar 2023 05:41:24 +0100 Subject: [PATCH] Update values --- tests/models/mgp_str/test_modeling_mgp_str.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/models/mgp_str/test_modeling_mgp_str.py b/tests/models/mgp_str/test_modeling_mgp_str.py index ab5b6422b627..990f78587da8 100644 --- 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, )