zhangyiss/mlx-examples
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx-examples.git synced 2025-08-29 07:30:06 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix test_generate

Browse Source
This commit is contained in:
L Lllvvuu 2024-12-27 01:53:15 -08:00
parent a28ca03e04
commit cded14988c
No known key found for this signature in database
GPG Key ID: CFAD5A25056DDD0F
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
llms/tests/test_generate.py
View File

@ -49,7 +49,7 @@ class TestGenerate(unittest.TestCase):
verbose=False,
logits_processors=[logits_processor],
)
self.assertEqual(len(all_toks), len(init_toks) + 5)
self.assertEqual(all_toks.shape[-1], len(init_toks) + 5)
if __name__ == "__main__":