Fix generation parameters test case

huggingface · Sep 16, 2023 · 76cbd7c · 76cbd7c
1 parent 7a8306f
commit 76cbd7c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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();