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

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -29,8 +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
- super().__init__(**kwargs)
34
  self.add_special_tokens({'additional_special_tokens':["<mask_1>"]})
35
 
36
  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(