Skip to content

Commit

Permalink
Update src/models.js
Browse files Browse the repository at this point in the history
  • Loading branch information
xenova authored Mar 7, 2024
1 parent 0efe69e commit 0c14c05
Showing 1 changed file with 1 addition and 10 deletions.
11 changes: 1 addition & 10 deletions src/models.js
Original file line number Diff line number Diff line change
Expand Up @@ -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 { }
//////////////////////////////////////////////////

//////////////////////////////////////////////////
Expand Down

0 comments on commit 0c14c05

Please sign in to comment.