Fix

mseeger · mseeger · commit 5cff6ce1dfa5 · 2025-07-05T13:51:29.000+02:00
diff --git a/tests/test_generate_speculatively.py b/tests/test_generate_speculatively.py
@@ -202,6 +202,7 @@ def test_main(fake_checkpoint_dir, monkeypatch, tensor_like):
                 ANY,
                 tensor_like,
                 53,
+                prompt_chunksize=16,
                 temperature=2.0,
                 top_k=2,
                 top_p=0.9,
