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 {}
