emanuelaboros commited on
Commit
dd2cf99
·
1 Parent(s): bdff5a0

checking if other model thigns are needed

Browse files
Files changed (1) hide show
  1. modeling_lang.py +2 -2
modeling_lang.py CHANGED
@@ -55,8 +55,8 @@ class ExtendedMultitaskModelForTokenClassification(PreTrainedModel):
55
 
56
  # print("Ignoring state_dict since model has no parameters.")
57
 
58
- def get_floret_model(self):
59
- return self.model_floret
60
 
61
  # def get_extended_attention_mask(
62
  # self, attention_mask, input_shape, device=None, dtype=torch.float
 
55
 
56
  # print("Ignoring state_dict since model has no parameters.")
57
 
58
+ # def get_floret_model(self):
59
+ # return self.model_floret
60
 
61
  # def get_extended_attention_mask(
62
  # self, attention_mask, input_shape, device=None, dtype=torch.float