Commit
·
dd2cf99
1
Parent(s):
bdff5a0
checking if other model thigns are needed
Browse files- 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 |
-
|
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
|