xcczach
/

test-model

Feature Extraction

Model card Files Files and versions

xcczach commited on Oct 8, 2024

Commit

2a0afa5

·

verified ·

1 Parent(s): 039788f

Update modeling_test.py

Files changed (1) hide show

modeling_test.py +1 -1

modeling_test.py CHANGED Viewed

@@ -16,7 +16,7 @@ class TestModel(PreTrainedModel):
             AutoConfig.from_pretrained("bert-base-uncased")
         )
         self.model3 = AutoModelForSequenceClassification.from_config(
-            AutoConfig.from_pretrained("bert-base-uncased", num_labels=2)
         )
     def forward(self, tensor):

             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):