Make sure custom configs work with Transformers (#15569)

* Make sure custom configs work with Transformers * Apply code review suggestions
2022-02-09 10:04:44 -05:00
parent 7732d0fe7a
commit 1f60bc46f3
5 changed files with 37 additions and 6 deletions
--- a/utils/test_module/custom_configuration.py
+++ b/utils/test_module/custom_configuration.py
@@ -7,3 +7,10 @@ class CustomConfig(PretrainedConfig):
    def __init__(self, attribute=1, **kwargs):
        self.attribute = attribute
        super().__init__(**kwargs)
+
+
+class NoSuperInitConfig(PretrainedConfig):
+    model_type = "custom"
+
+    def __init__(self, attribute=1, **kwargs):
+        self.attribute = attribute