add model

config.json

@@ -0,0 +1,14 @@
+{
+  "architectures": [
+    "AwesomeModel"
+  ],
+  "attribute": 1,
+  "auto_map": {
+    "AutoConfig": "config.AwesomeConfig",
+    "AutoModel": "model.AwesomeModel"
+  },
+  "hidden_size": 16,
+  "model_type": "awesome",
+  "torch_dtype": "float32",
+  "transformers_version": "4.17.0.dev0"
+}

config.py

@@ -0,0 +1,10 @@
+from transformers import PretrainedConfig
+
+
+class AwesomeConfig(PretrainedConfig):
+    model_type = "awesome"
+
+    def __init__(self, attribute=1, **kwargs):
+        self.attribute = attribute
+        self.hidden_size = 16
+        super().__init__(**kwargs)

model.py

@@ -0,0 +1,17 @@
+import torch
+
+from transformers import PreTrainedModel
+
+from .config import AwesomeConfig
+
+
+class AwesomeModel(PreTrainedModel):
+    config_class = AwesomeConfig
+    base_model_prefix = "base"
+
+    def __init__(self, config):
+        super().__init__(config)
+        self.linear = torch.nn.Linear(config.hidden_size, config.hidden_size)
+
+    def forward(self, x):
+        return self.linear(x)

pytorch_model.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:37ab646b69a0f3771e34a402e9b85bc5d5124ea888d4c06d56103adbf4d98241
+size 2243