use torch.testing.assertclose instead to get more details about error in cis (#35659)
* use torch.testing.assertclose instead to get more details about error in cis * fix * style * test_all * revert for I bert * fixes and updates * more image processing fixes * more image processors * fix mamba and co * style * less strick * ok I won't be strict * skip and be done * up
This commit is contained in:
@@ -250,7 +250,7 @@ class Mamba2ModelTest(ModelTesterMixin, GenerationTesterMixin, PipelineTesterMix
|
||||
if "D" in name:
|
||||
if param.requires_grad:
|
||||
# check if it's a ones like
|
||||
self.assertTrue(torch.allclose(param.data, torch.ones_like(param.data), atol=1e-5, rtol=1e-5))
|
||||
torch.testing.assert_close(param.data, torch.ones_like(param.data), rtol=1e-5, atol=1e-5)
|
||||
|
||||
@unittest.skip(reason="Mamba 2 weights are not tied")
|
||||
def test_tied_weights_keys(self):
|
||||
@@ -439,4 +439,4 @@ class Mamba2IntegrationTest(unittest.TestCase):
|
||||
mixer.eval()
|
||||
out_eval = mixer(hidden_states)
|
||||
|
||||
self.assertTrue(torch.allclose(out_train, out_eval, atol=1e-3))
|
||||
torch.testing.assert_close(out_train, out_eval, rtol=1e-3, atol=1e-3)
|
||||
|
||||
Reference in New Issue
Block a user