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

Fix ESM LM head test (#20045)

Browse Source 
* Fix esm lm head test

* make fixup
This commit is contained in:
Matt
2022-11-04 12:45:34 +00:00
committed by GitHub
parent d447c460b1
commit 1076d587b5
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tests/models/esm/test_modeling_esm.py
View File

@@ -285,7 +285,7 @@ class EsmModelIntegrationTest(TestCasePlus):
self.assertEqual(output.shape, expected_shape)
expected_slice = torch.tensor(
[[[15.0973, -6.6406, -1.1351], [-0.2209, -9.9622, 4.2109], [-1.6055, -10.0023, 1.5914]]]
[[[8.9215, -10.5898, -6.4671], [-6.3967, -13.9114, -1.1212], [-7.7812, -13.9516, -3.7406]]]
)
self.assertTrue(torch.allclose(output[:, :3, :3], expected_slice, atol=1e-4))