nb-roberta-base-scandinavian / create_config.py
pere's picture
moved model from pere/nb-bert-roberta-base-scandinavian-long
c539a6e
from transformers import RobertaConfig
model_dir = "./" # ${MODEL_DIR}
config = RobertaConfig.from_pretrained("roberta-base")
config.save_pretrained(model_dir)