from transformers import Lfm2PreTrainedModel from transformers.modeling_layers import GenericForTokenClassification class Lfm2ForTokenClassification(GenericForTokenClassification,Lfm2PreTrainedModel): pass