From d02f0780fcd55cc0d94d733319b70813c7fbcdc1 Mon Sep 17 00:00:00 2001
From: Joshua Lochner <admin@xenova.com>
Date: Wed, 6 Mar 2024 17:39:36 +0200
Subject: [PATCH] Naming nits

---
 src/models.js | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/src/models.js b/src/models.js
index d4a44525b..f5a930e35 100644
--- a/src/models.js
+++ b/src/models.js
@@ -5318,9 +5318,9 @@ export class SegformerForSemanticSegmentation extends SegformerPreTrainedModel {
 
 //////////////////////////////////////////////////
 // StableLm models
-export class StableLMPreTrainedModel extends PreTrainedModel {
+export class StableLmPreTrainedModel extends PreTrainedModel {
     /**
-     * Creates a new instance of the `StableLMPreTrainedModel` class.
+     * Creates a new instance of the `StableLmPreTrainedModel` class.
      * @param {Object} config The configuration of the model.
      * @param {any} session The ONNX session containing the model weights.
      * @param {GenerationConfig} generation_config The generation configuration.
@@ -5339,9 +5339,9 @@ export class StableLMPreTrainedModel extends PreTrainedModel {
 }
 
 /**
- * The bare StableLM Model transformer outputting raw hidden-states without any specific head on top.
+ * The bare StableLm Model transformer outputting raw hidden-states without any specific head on top.
  */
-export class StableLMModel extends StableLMPreTrainedModel { }
+export class StableLmModel extends StableLmPreTrainedModel { }
 
 /**
  * StableLm Model with a `language modeling` head on top for Causal Language Modeling (with past).