emanuelaboros
/

lang-detect

Token Classification

language-identification

Model card Files Files and versions Community

emanuelaboros commited on Feb 27

Commit

52d99b3

·

1 Parent(s): 290a993

testin the trick

Files changed (1) hide show

modeling_stacked.py +2 -2

modeling_stacked.py CHANGED Viewed

@@ -53,8 +53,8 @@ class ExtendedMultitaskModelForTokenClassification(PreTrainedModel):
         # Load floret model
         self.dummy_param = nn.Parameter(torch.zeros(1))
-        model_floret = floret.load_model(self.config.filename)
-        self.model_floret = SafeFloretWrapper(model_floret)
         # input_ids = "this is a text"
         # predictions, probabilities = self.model_floret.predict([input_ids], k=1)

         # Load floret model
         self.dummy_param = nn.Parameter(torch.zeros(1))
+        self.model_floret = floret.load_model(self.config.filename)
+        # self.model_floret = SafeFloretWrapper(model_floret)
         # input_ids = "this is a text"
         # predictions, probabilities = self.model_floret.predict([input_ids], k=1)