from transformers import AutoConfig, AutoModel from .configuration_maskrcnn import MaskRCNNConfig from .modeling_maskrcnn import MaskRCNNForInstanceSegmentation AutoConfig.register("maskrcnn", MaskRCNNConfig) AutoModel.register(MaskRCNNConfig, MaskRCNNForInstanceSegmentation)