huggingface/text-generation-inference
5
0
Fork 0
mirror of https://github.com/huggingface/text-generation-inference.git synced 2025-09-10 20:04:52 +00:00
Code Issues Packages Projects Releases Wiki Activity

Revert test change.

Browse Source
This commit is contained in:
Nicolas Patry 2023-09-26 16:46:06 +02:00
parent 8a16c48595
commit 6378303d23
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
server/tests/models/test_seq2seq_lm.py
View File

@ -168,7 +168,7 @@ def test_seq2seq_lm_generate_token_completion(
assert next_batch is None
assert len(generations) == 1
assert generations[0].generated_text.text == " a few weeks"
assert generations[0].generated_text.text == "a few weeks"
assert generations[0].request_id == default_seq2seq_lm_batch.requests[0].id
assert generations[0].generated_text.generated_tokens == 7
@ -186,7 +186,7 @@ def test_seq2seq_lm_generate_token_completion_multi(
assert next_batch is not None
assert len(generations) == 2
assert generations[1].generated_text.text == " a few "
assert generations[1].generated_text.text == "a few "
assert (
generations[1].request_id
== default_multi_requests_seq2seq_lm_batch.requests[1].id