We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
test_compare_unprocessed_logit_scores
1 parent 58c7689 commit 23b7e73Copy full SHA for 23b7e73
tests/generation/test_utils.py
@@ -3807,7 +3807,7 @@ def test_compare_unprocessed_logit_scores(self):
3807
logits_gen = outputs.logits[0][0]
3808
3809
# assert that unprocessed logits from generate() are same as those from modal eval()
3810
- self.assertListEqual(logits_fwd.tolist(), logits_gen.tolist())
+ torch.testing.assert_allclose(logits_fwd.tolist(), logits_gen.tolist())
3811
3812
def test_return_unprocessed_logit_scores(self):
3813
# tell model to generate text and return unprocessed/unwarped logit scores
0 commit comments