Skip to content

GitLab

"git@developer.sourcefind.cn:chenpangpang/transformers.git" did not exist on "0199a484ebaeac4492693b3f49626b7c220488b4"
Unverified Commit 52a57f7c authored by Yih-Dar's avatar Yih-Dar Committed by GitHub
Browse files

Update expected values in `MgpstrModelIntegrationTest` (#22195) 



Update values
Co-authored-by: default avatarydshieh <ydshieh@users.noreply.github.com>
parent 1485bd9c
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
tests/models/mgp_str/test_modeling_mgp_str.py
View file @ 52a57f7c
...@@ -262,7 +262,7 @@ class MgpstrModelIntegrationTest(unittest.TestCase): ...@@ -262,7 +262,7 @@ class MgpstrModelIntegrationTest(unittest.TestCase):
self.assertEqual(out_strs["generated_text"][0], expected_text) self.assertEqual(out_strs["generated_text"][0], expected_text)
expected_slice = torch.tensor( 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, device=torch_device,
) )
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment