dongxiaoqun
commited on
Commit
·
7cdad11
1
Parent(s):
bef8614
Update README.md
Browse files
README.md
CHANGED
@@ -29,7 +29,8 @@ from transformers import PegasusForConditionalGeneration,BertTokenizer
|
|
29 |
class PegasusTokenizer(BertTokenizer):
|
30 |
model_input_names = ["input_ids", "attention_mask"]
|
31 |
def __init__(self, **kwargs):
|
32 |
-
super().__init__(pre_tokenizer=lambda x: jieba.cut(x, HMM=False), **kwargs)
|
|
|
33 |
self.add_special_tokens({'additional_special_tokens':["<mask_1>"]})
|
34 |
|
35 |
def build_inputs_with_special_tokens(
|
|
|
29 |
class PegasusTokenizer(BertTokenizer):
|
30 |
model_input_names = ["input_ids", "attention_mask"]
|
31 |
def __init__(self, **kwargs):
|
32 |
+
# super().__init__(pre_tokenizer=lambda x: jieba.cut(x, HMM=False), **kwargs)
|
33 |
+
super().__init__(**kwargs)
|
34 |
self.add_special_tokens({'additional_special_tokens':["<mask_1>"]})
|
35 |
|
36 |
def build_inputs_with_special_tokens(
|