@huggingface/transformers
Version:
State-of-the-art Machine Learning for the web. Run 🤗 Transformers directly in your browser, with no need for a server!
12 lines (8 loc) • 456 B
JavaScript
import { PreTrainedModel } from '../modeling_utils.js';
export class CohereAsrPreTrainedModel extends PreTrainedModel {
requires_attention_mask = false;
main_input_name = 'input_features';
forward_params = ['input_features', 'decoder_input_ids', 'decoder_attention_mask', 'past_key_values'];
}
export class CohereAsrModel extends CohereAsrPreTrainedModel {}
export class CohereAsrForConditionalGeneration extends CohereAsrPreTrainedModel {}