emanuelaboros
/

lang-detect

Token Classification

language-identification

Model card Files Files and versions

emanuelaboros commited on Feb 27, 2025

Commit

2bc2b78

·

1 Parent(s): f729b09

testin the trick

Files changed (1) hide show

modeling_stacked.py +1 -1

modeling_stacked.py CHANGED Viewed

@@ -102,7 +102,7 @@ class ExtendedMultitaskModelForTokenClassification(PreTrainedModel):
         print("Ignoring state_dict since model has no parameters.")
     def get_floret_model(self):
-        return self.model_floret
     def get_extended_attention_mask(
         self, attention_mask, input_shape, device=None, dtype=torch.float

         print("Ignoring state_dict since model has no parameters.")
     def get_floret_model(self):
+        return self.safe_floret
     def get_extended_attention_mask(
         self, attention_mask, input_shape, device=None, dtype=torch.float