Skip to content

Commit

Permalink
Naming nits
Browse files Browse the repository at this point in the history
  • Loading branch information
xenova authored Mar 6, 2024
1 parent 295c317 commit d02f078
Showing 1 changed file with 4 additions and 4 deletions.
8 changes: 4 additions & 4 deletions src/models.js
Original file line number Diff line number Diff line change
Expand Up @@ -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.
Expand All @@ -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).
Expand Down

0 comments on commit d02f078

Please sign in to comment.