new version

config.json

@@ -3,7 +3,7 @@
     "MNIST_Classifier"
   ],
   "auto_map": {
-    "AutoConfig": "config.Config",
+    "AutoConfig": "config.MNIST_config",
     "AutoModel": "model.MNIST_Classifier"
   },
   "hidden_size1": 128,

config.py

@@ -1,5 +1,5 @@
 from transformers import PretrainedConfig
-class Config(PretrainedConfig):
+class MNIST_config(PretrainedConfig):
     model_type = "MNIST_Classifier"
     def __init__(self, **kwargs):
         super().__init__(**kwargs)

model.py

@@ -1,13 +1,12 @@
 ###import os,sys
 ###sys.path.insert(1,os.path.join(sys.path[0],".."))
 from network import Net
-from config import Config
+from config import MNIST_config
 from transformers import PreTrainedModel
 # utils not used but importing it forces the upload to huggingface hub to include it
 
-
 class MNIST_Classifier(PreTrainedModel):
-    config_class = Config
+    config_class = MNIST_config
     def __init__(self, config):
         super().__init__(config)
         self.classifier=Net(config.input_size,config.hidden_size1,config.hidden_size2,