From d8bef274ee0247ab047520edc3977c4239c17148 Mon Sep 17 00:00:00 2001
From: ocavue <ocavue@gmail.com>
Date: Sun, 8 Dec 2024 10:43:20 +1100
Subject: [PATCH] Add types for MultiModalityCausalLM

---
 src/models.js | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/src/models.js b/src/models.js
index 7ab8a3302..fe4d46365 100644
--- a/src/models.js
+++ b/src/models.js
@@ -1537,7 +1537,6 @@ export class PreTrainedModel extends Callable {
 
     /**
      * Generates sequences of token ids for models with a language modeling head.
-     // @ts-expect-error TS2306
      * @param {import('./generation/parameters.js').GenerationFunctionParameters} options
      * @returns {Promise<ModelOutput|Tensor>} The output of the model, which can contain the generated token ids, attentions, and scores.
      */
@@ -6575,8 +6574,10 @@ export class MultiModalityCausalLM extends MultiModalityPreTrainedModel {
         'past_key_values',
     ];
 
+    /**
+     * @param {ConstructorParameters<typeof MultiModalityPreTrainedModel>} args
+     */
     constructor(...args) {
-        // @ts-expect-error TS2556
         super(...args);
 
         // State-based approach to switch out which heads to use during generation