From 76cbd7c73bd1e20b592b317565b5b7d8b056e743 Mon Sep 17 00:00:00 2001
From: Joshua Lochner <admin@xenova.com>
Date: Sun, 17 Sep 2023 01:30:58 +0200
Subject: [PATCH] Fix generation parameters test case

---
 tests/generation.test.js | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tests/generation.test.js b/tests/generation.test.js
index 6a34e496e..43e18675e 100644
--- a/tests/generation.test.js
+++ b/tests/generation.test.js
@@ -68,7 +68,7 @@ describe('Generation parameters', () => {
             });
 
             const tokens = generator.tokenizer.encode(outputs[0])
-            expect(tokens.length).toBeGreaterThanOrEqual(MIN_NEW_TOKENS + 1); // + 1 due to forced BOS token
+            expect(tokens.length).toBeGreaterThanOrEqual(MIN_NEW_TOKENS);
         }
 
         await generator.dispose();