custom_modeling.py

from transformers import AutoConfig, AutoModel
from modeling_cnn_ner import CNNForNER

def get_model(pretrained_model_name_or_path):
    config = AutoConfig.from_pretrained(pretrained_model_name_or_path)
    model = CNNForNER(
        pretrained_model_name=config.pretrained_model_name,
        num_classes=config.num_classes,
        max_length=config.max_length
    )
    return model