"tests/vscode:/vscode.git/clone" did not exist on "914289ac4b7994507fa7329bf6f54572b32ae061"
Unverified Commit 1076d587 authored by Matt's avatar Matt Committed by GitHub
Browse files

Fix ESM LM head test (#20045)

* Fix esm lm head test

* make fixup
parent d447c460
...@@ -285,7 +285,7 @@ class EsmModelIntegrationTest(TestCasePlus): ...@@ -285,7 +285,7 @@ class EsmModelIntegrationTest(TestCasePlus):
self.assertEqual(output.shape, expected_shape) self.assertEqual(output.shape, expected_shape)
expected_slice = torch.tensor( 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)) self.assertTrue(torch.allclose(output[:, :3, :3], expected_slice, atol=1e-4))
......
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