wehub-resource-sync
14 lines
475 B
JavaScript
14 lines
475 B
JavaScript
import { PreTrainedModel } from '../modeling_utils.js';
|
|
import { SequenceClassifierOutput } from '../modeling_outputs.js';
|
|
|
|
export class BeitPreTrainedModel extends PreTrainedModel {}
|
|
export class BeitModel extends BeitPreTrainedModel {}
|
|
export class BeitForImageClassification extends BeitPreTrainedModel {
|
|
/**
|
|
* @param {any} model_inputs
|
|
*/
|
|
async _call(model_inputs) {
|
|
return new SequenceClassifierOutput(await super._call(model_inputs));
|
|
}
|
|
}
|