Config to Model mapping

2020-01-13 20:05:20 +00:00
parent cf8a70bf68
commit b803b067bf
17 changed files with 58 additions and 55 deletions
--- a/templates/adding_a_new_model/configuration_xxx.py
+++ b/templates/adding_a_new_model/configuration_xxx.py
@@ -58,6 +58,7 @@ class XxxConfig(PretrainedConfig):
            layer_norm_eps: The epsilon used by LayerNorm.
    """
    pretrained_config_archive_map = XXX_PRETRAINED_CONFIG_ARCHIVE_MAP
+    model_type = "xxx"

    def __init__(
        self,