Upload model

config.json

@@ -0,0 +1,13 @@
+{
+  "architectures": [
+    "METLModel"
+  ],
+  "auto_map": {
+    "AutoConfig": "huggingface_wrapper.METLConfig",
+    "AutoModel": "huggingface_wrapper.METLModel"
+  },
+  "id": null,
+  "model_type": "METL",
+  "torch_dtype": "float32",
+  "transformers_version": "4.42.4"
+}

huggingface_wrapper.py

@@ -0,0 +1,58 @@
+from transformers import PretrainedConfig, PreTrainedModel
+import metl.models as models
+from metl.main import *
+from metl.test_cnn import MNIST_CNN
+
+class METLConfig(PretrainedConfig):
+    IDENT_UUID_MAP = IDENT_UUID_MAP
+    UUID_URL_MAP = UUID_URL_MAP
+    model_type = "METL"
+
+    def __init__(
+            self,
+            id:str = None,
+            **kwargs,
+    ):
+        self.id = id
+        super().__init__(**kwargs)
+
+#   input_shape,
+#             num_layers,
+#             conv_kernel_sizes,
+#             conv_channel_sizes,
+#             pool_kernel_sizes
+
+class METLModel(PreTrainedModel):
+    config_class = METLConfig
+    def __init__(self, config:METLConfig):
+        super().__init__(config)
+        self.model = MNIST_CNN(
+            input_shape=(28, 28),
+            num_layers = 4,
+            conv_kernel_sizes = [4, 3, 3, 3],
+            conv_channel_sizes = [(1, 16), (16, 32), (32, 64), (64, 128)],
+            pool_kernel_sizes = [2, 2, 2]
+            )
+        self.encoder = None
+        self.config = config
+        
+    def forward(self, X, pdb_fn=None):
+        if pdb_fn:
+            return self.model(X, pdb_fn=pdb_fn)
+        return self.model(X)
+    
+    def load_from_uuid(self, id):
+        if id:
+            id = id.lower()
+            assert id in self.config.UUID_URL_MAP, "ID given does not reference a valid METL model in the IDENT_UUID_MAP"
+            self.config.id = id
+
+        self.model, self.encoder = get_from_uuid(self.config.UUID_URL_MAP[self.config.id])
+
+    def load_from_indent(self, id):
+        if id:
+            id = id.lower()
+            assert id in self.config.IDENT_UUID_MAP, "ID given does not reference a valid METL model in the IDENT_UUID_MAP"
+            self.config.id = id
+
+        self.model, self.encoder = get_from_ident(self.config.IDENT_UUID_MAP[self.config.id])

model.safetensors

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