SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

Update expected values in MgpstrModelIntegrationTest (#22195)

Browse Source 
Update values

Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
This commit is contained in:
Yih-Dar
2023-03-16 12:48:52 +01:00
committed by GitHub
parent 1485bd9c02
commit 52a57f7c7c
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tests/models/mgp_str/test_modeling_mgp_str.py
View File

@@ -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,
) )