Fix model weights

model.py

@@ -25,7 +25,7 @@ class BertMesh(PreTrainedModel):
         dropout=0,
         multilabel_attention=False,
     ):
-        super().__init__(config=AutoConfig.from_pretrained(pretrained_model))
+        super().__init__(config=config)
         self.config.auto_map = {"AutoModel": "transformers_model.BertMesh"}
         self.pretrained_model = pretrained_model
         self.num_labels = num_labels

pytorch_model.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:92fc6bad6bb26250dc89778a62810709e103dcb3992a13e216d192d9cb398127
-size 647445539
+oid sha256:9588cca776ff05298d33d4785a1053cafddc231e3ebc069bc8137ed28c647fa0
+size 647426403