new version

config.json

@@ -0,0 +1,16 @@
+{
+  "architectures": [
+    "MNIST_Classifier"
+  ],
+  "auto_map": {
+    "AutoConfig": "config.Config",
+    "AutoModel": "model.MNIST_Classifier"
+  },
+  "hidden_size1": 128,
+  "hidden_size2": 64,
+  "input_size": 784,
+  "model_type": "MNIST_Classifier",
+  "output_size": 10,
+  "torch_dtype": "float32",
+  "transformers_version": "4.35.2"
+}

config.py

@@ -0,0 +1,14 @@
+from transformers import PretrainedConfig
+class Config(PretrainedConfig):
+    model_type = "MNIST_Classifier"
+    def __init__(self, **kwargs):
+        super().__init__(**kwargs)
+  
+        for key,value in kwargs.items():
+            setattr(self,key,value)
+        #print(key,value)
+        #self.input_size=kwargs['input_size']
+        #self.hidden_size1=kwargs["hidden_size1"]
+        #self.hidden_size2=kwargs["hidden_size2"]
+        #self.output_size=kwargs["output_size"]
+

model.py

@@ -0,0 +1,17 @@
+###import os,sys
+###sys.path.insert(1,os.path.join(sys.path[0],".."))
+from network import Net
+from config import 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
+    def __init__(self, config):
+        super().__init__(config)
+        self.classifier=Net(config.input_size,config.hidden_size1,config.hidden_size2,
+           config.output_size)
+
+    def forward(self, input):
+        return self.classifier(input)

model.safetensors

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