from transformers import PreTrainedConfig class ALM2VecConfig(PreTrainedConfig): model_type = "alm2vec"