ALM2Vec-FT / configuration_alm2vec.py
frankie137's picture
Add files using upload-large-folder tool
b94bf59 verified
Raw History Blame Contribute Delete
110 Bytes
from transformers import PreTrainedConfig
class ALM2VecConfig(PreTrainedConfig):
model_type = "alm2vec"