Update configuration_auto.py

This commit is contained in:
Stella Biderman 2021-08-05 16:21:44 -04:00 committed by GitHub
parent cfaaae4c4d
commit e9860e9c04
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -204,6 +204,7 @@ CONFIG_MAPPING = OrderedDict(
("mobilebert", MobileBertConfig),
("mpnet", MPNetConfig),
("mt5", MT5Config),
("openai-gpt", OpenAIGPTConfig),
("pegasus", PegasusConfig),
("prophetnet", ProphetNetConfig),
("rag", RagConfig),