@huggingface/transformers
Version:
State-of-the-art Machine Learning for the web. Run 🤗 Transformers directly in your browser, with no need for a server!
14 lines (12 loc) • 490 B
JavaScript
import { PreTrainedModel } from '../modeling_utils.js';
import { RTDetrObjectDetectionOutput } from '../rt_detr/modeling_rt_detr.js';
export class DFinePreTrainedModel extends PreTrainedModel {}
export class DFineModel extends DFinePreTrainedModel {}
export class DFineForObjectDetection extends DFinePreTrainedModel {
/**
* @param {any} model_inputs
*/
async _call(model_inputs) {
return new RTDetrObjectDetectionOutput(await super._call(model_inputs));
}
}