Upload config

MyConfig.py

@@ -0,0 +1,17 @@
+
+from transformers import PretrainedConfig
+from typing import List
+
+class MnistConfig(PretrainedConfig):
+    # since we have an image classification task
+    # we need to put a model type that is close to our task
+    # don't worry this will not affect our model
+    model_type = "MobileNetV1"
+    def __init__(
+        self,
+        conv1=10,
+        conv2=20,
+        **kwargs):
+      self.conv1 = conv1
+      self.conv2 = conv2
+      super().__init__(**kwargs)

config.json

@@ -0,0 +1,9 @@
+{
+  "auto_map": {
+    "AutoConfig": "MyConfig.MnistConfig"
+  },
+  "conv1": 10,
+  "conv2": 20,
+  "model_type": "MobileNetV1",
+  "transformers_version": "4.35.2"
+}