Upload model

modeling_test.py

@@ -12,9 +12,9 @@ class TestModel(PreTrainedModel):
         super().__init__(config)
         self.input_dim = config.input_dim
         self.model1 = nn.Linear(config.input_dim, config.output_dim)
-        self.model2 = AutoModelForMaskedLM.from_config(
-            AutoConfig.from_pretrained("bert-base-uncased")
-        )
+        # self.model2 = AutoModelForMaskedLM.from_config(
+        #     AutoConfig.from_pretrained("bert-base-uncased")
+        # )
 
     def forward(self, tensor):
         return self.model1(tensor)

pytorch_model.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:038eef714a985a0277794c8e1e54df28739eef3053a4c7af6aad87774bdde0bf
-size 438125501
+oid sha256:652bf7b2bdc254841672719f7491681582fbb546a9d05fe470da0d4315f2d192
+size 1215