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

modify integration tests

Browse Source
This commit is contained in:
OlivierDehaene 2023-01-30 16:32:44 +01:00
parent f8e230f65c
commit 614a1a7202
2 changed files with 8 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
launcher/tests/bloom_560m.json
View File

@ -3,7 +3,7 @@
"details": { "details": {
"finish_reason": "length", "finish_reason": "length",
"generated_tokens": 20, "generated_tokens": 20,
"tokens": [ "prefill": [
[ [
10264, 10264,
"Test", "Test",
@ -13,7 +13,9 @@
8821, 8821,
" request", " request",
-11.895094 -11.895094
], ]
],
"tokens": [
[ [
17, 17,
".", ".",

6
launcher/tests/mt0_base.json
View File

@ -3,12 +3,14 @@
"details": { "details": {
"finish_reason": "length", "finish_reason": "length",
"generated_tokens": 20, "generated_tokens": 20,
"tokens": [ "prefill": [
[ [
0, 0,
"<pad>", "<pad>",
null null
], ]
],
"tokens": [
[ [
259, 259,
"", "",