From 0c14c050814cadc7e95c8d470b8530f6d286b383 Mon Sep 17 00:00:00 2001
From: Joshua Lochner <admin@xenova.com>
Date: Thu, 7 Mar 2024 03:03:53 +0200
Subject: [PATCH] Update src/models.js

---
 src/models.js | 11 +----------
 1 file changed, 1 insertion(+), 10 deletions(-)

diff --git a/src/models.js b/src/models.js
index f5a930e35..ef3d2806c 100644
--- a/src/models.js
+++ b/src/models.js
@@ -5346,16 +5346,7 @@ export class StableLmModel extends StableLmPreTrainedModel { }
 /**
  * StableLm Model with a `language modeling` head on top for Causal Language Modeling (with past).
  */
-export class StableLmForCausalLM extends StableLmPreTrainedModel {
-    /**
-     * Calls the model on new inputs.
-     * @param {Object} model_inputs The inputs to the model.
-     * @returns {Promise<CausalLMOutput>} An object containing the model's output logits for causal language modeling.
-     */
-    async _call(model_inputs) {
-        return new CausalLMOutputWithPast(await super._call(model_inputs));
-    }
-}
+export class StableLmForCausalLM extends StableLmPreTrainedModel { }
 //////////////////////////////////////////////////
 
 //////////////////////////////////////////////////