Skip to content

Commit

Permalink
test fix WIP
Browse files Browse the repository at this point in the history
  • Loading branch information
tengomucho committed Feb 28, 2024
1 parent 9080f68 commit 45a4a77
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion text-generation-inference/tests/test_generator.py
Original file line number Diff line number Diff line change
Expand Up @@ -117,7 +117,7 @@ def test_prefill(input_text, token_id, token_text, do_sample, batch_size, model_
[
"It was a bright cold day in April, and the clocks were striking thirteen.",
20,
" All was silent, and it seemed, as was the custom in the United States, that the day",
" We sat outside the house, drinking coffee, listening to the traffic. And then, suddenly, we",
True,
],
],
Expand Down

0 comments on commit 45a4a77

Please sign in to comment.