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

fix test

Browse Source
This commit is contained in:
OlivierDehaene 2023-04-13 16:52:12 +02:00
parent 42196a1af8
commit c7803c41e8
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
clients/python/tests/test_client.py
View File

@ -17,7 +17,7 @@ def test_generate(flan_t5_xxl_url, hf_headers):
assert response.details.prefill[0] == PrefillToken(id=0, text="<pad>", logprob=None)
assert len(response.details.tokens) == 1
assert response.details.tokens[0] == Token(
id=3, text=" ", logprob=-1.984375, special=False
id=3, text="", logprob=-1.984375, special=False
)
@ -83,7 +83,7 @@ async def test_generate_async(flan_t5_xxl_url, hf_headers):
assert response.details.prefill[0] == PrefillToken(id=0, text="<pad>", logprob=None)
assert len(response.details.tokens) == 1
assert response.details.tokens[0] == Token(
id=3, text=" ", logprob=-1.984375, special=False
id=3, text="", logprob=-1.984375, special=False
)