Skip to content

Commit

Permalink
fix(test): Adapt test for CI/CD
Browse files Browse the repository at this point in the history
Pytorch's RNG seems to behave differently in the docker image with
Pytorch XLA 2.3.0. This leads to different results when using
do_sample=True, so tests have been updated accordingly.
  • Loading branch information
tengomucho committed Feb 28, 2024
1 parent 9c5afba commit aef6bb3
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions text-generation-inference/tests/test_generator.py
Original file line number Diff line number Diff line change
Expand Up @@ -75,8 +75,8 @@ def create_request(
],
[
"It was a bright cold day in April, and the clocks were striking thirteen.",
1439,
" All",
775,
" We",
True,
],
],
Expand Down 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 aef6bb3

Please sign in to comment.