mirror of
https://github.com/ml-explore/mlx-examples.git
synced 2025-09-10 21:37:45 +08:00
refactor sampler/processor and a few improvements
This commit is contained in:
@@ -46,7 +46,7 @@ class TestGenerate(unittest.TestCase):
|
||||
"hello",
|
||||
max_tokens=5,
|
||||
verbose=False,
|
||||
logits_processor=[logits_processor],
|
||||
logits_processors=[logits_processor],
|
||||
)
|
||||
self.assertEqual(len(all_toks), len(init_toks) + 5)
|
||||
|
||||
|
Reference in New Issue
Block a user