dongxiaoqun commited on
Commit
7cdad11
·
1 Parent(s): bef8614

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -1
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(