From ed748834d6bdaefed4f3858b2c17effb35722bf8 Mon Sep 17 00:00:00 2001
From: Joshua Lochner <admin@xenova.com>
Date: Sat, 6 Apr 2024 21:16:08 +0200
Subject: [PATCH] Fix JSDoc

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

diff --git a/src/generation/stopping_criteria.js b/src/generation/stopping_criteria.js
index c9863bca8..c343d3faa 100644
--- a/src/generation/stopping_criteria.js
+++ b/src/generation/stopping_criteria.js
@@ -116,7 +116,7 @@ export class EosTokenCriteria extends StoppingCriteria {
      * 
      * @param {number[][]} input_ids 
      * @param {number[][]} scores 
-     * @returns 
+     * @returns {boolean[]}
      */
     _call(input_ids, scores) {
         return input_ids.map(ids => this.eos_token_id.includes(ids.at(-1)));