Upload model

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:ff5ae4b2762b833251833bd2023a1cac7cd6d2e078e1693a73e34dcb4edacfc8
+oid sha256:d1349aa18008de4bbf6906e998320035af0d3f8f160228bb3ce19251e986acf4
 size 876042724

modeling_test.py

@@ -1,23 +1,23 @@
-from transformers import PreTrainedModel
-from .configuration_test import TestConfig
-import torch.nn as nn
-from transformers import AutoModelForMaskedLM, AutoConfig
-from transformers import AutoModelForSequenceClassification
-
-
-class TestModel(PreTrainedModel):
-    config_class = TestConfig
-
-    def __init__(self, config: TestConfig):
-        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.model3 = AutoModelForSequenceClassification.from_config(
-            AutoConfig.from_pretrained("google-bert/bert-base-uncased", num_labels=2)
-        )
-
-    def forward(self, tensor):
-        return self.model1(tensor)
+from transformers import PreTrainedModel
+from .configuration_test import TestConfig
+import torch.nn as nn
+from transformers import AutoModelForMaskedLM, AutoConfig
+from transformers import AutoModelForSequenceClassification
+
+
+class TestModel(PreTrainedModel):
+    config_class = TestConfig
+
+    def __init__(self, config: TestConfig):
+        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.model3 = AutoModelForSequenceClassification.from_config(
+            AutoConfig.from_pretrained("google-bert/bert-base-uncased", num_labels=2)
+        )
+
+    def forward(self, tensor):
+        return self.model1(tensor)