Fix JSDoc

huggingface · Apr 6, 2024 · ed74883 · ed74883
1 parent b88b6a2
commit ed74883
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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)));