xcczach
/

test-model

Feature Extraction

Model card Files Files and versions Community

xcczach commited on 14 days ago

Commit

f1799a1

•

1 Parent(s): def78c6

Upload model

Files changed (2) hide show

modeling_test.py +3 -3
pytorch_model.bin +2 -2

modeling_test.py CHANGED Viewed

@@ -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")
-        # )
     def forward(self, tensor):
         return self.model1(tensor)

         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("albert/albert-base-v2")
+        )
     def forward(self, tensor):
         return self.model1(tensor)

pytorch_model.bin CHANGED Viewed

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:28f2f2ab50560db6d5ac24d702ad6c2d50aba3c04cc11034a559380394b068e3
-size 1215

 version https://git-lfs.github.com/spec/v1
+oid sha256:0f88faa6c688ec9ab7b59910120c1e3bf97817b8d41a68104f608db5b2830260
+size 44897099