From 614a1a7202d5839aba0f23c0566299df6a7708ff Mon Sep 17 00:00:00 2001 From: OlivierDehaene <23298448+OlivierDehaene@users.noreply.github.com> Date: Mon, 30 Jan 2023 16:32:44 +0100 Subject: [PATCH] modify integration tests --- launcher/tests/bloom_560m.json | 6 ++++-- launcher/tests/mt0_base.json | 6 ++++-- 2 files changed, 8 insertions(+), 4 deletions(-) diff --git a/launcher/tests/bloom_560m.json b/launcher/tests/bloom_560m.json index d17f1ed4..a81d1982 100644 --- a/launcher/tests/bloom_560m.json +++ b/launcher/tests/bloom_560m.json @@ -3,7 +3,7 @@ "details": { "finish_reason": "length", "generated_tokens": 20, - "tokens": [ + "prefill": [ [ 10264, "Test", @@ -13,7 +13,9 @@ 8821, " request", -11.895094 - ], + ] + ], + "tokens": [ [ 17, ".", diff --git a/launcher/tests/mt0_base.json b/launcher/tests/mt0_base.json index 1b772282..51cb8b5c 100644 --- a/launcher/tests/mt0_base.json +++ b/launcher/tests/mt0_base.json @@ -3,12 +3,14 @@ "details": { "finish_reason": "length", "generated_tokens": 20, - "tokens": [ + "prefill": [ [ 0, "", null - ], + ] + ], + "tokens": [ [ 259, "",