Upload model

huggingface_wrapper.py

@@ -1,7 +1,6 @@
 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
@@ -16,23 +15,11 @@ class METLConfig(PretrainedConfig):
         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.model = None
         self.encoder = None
         self.config = config
         

model.safetensors

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