Skip to content

Commit

Permalink
fix(CI): adapt expected result in do_sample test
Browse files Browse the repository at this point in the history
The test produces different results after some operations are being done
in a slightly different order.
  • Loading branch information
tengomucho committed Mar 14, 2024
1 parent 27a2669 commit be1194b
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 @@ -109,7 +109,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,
" We sat outside the house, drinking coffee, listening to the traffic. And then, suddenly, we",
" We sat outside the house, drinking coffee, listening to the orchestra playing through the window. We could",
True,
],
],
Expand Down

0 comments on commit be1194b

Please sign in to comment.