stefan-insilico commited on
Commit
4a371ff
1 Parent(s): de14dfb

Update precious3_gpt_multi_modal.py

Browse files
Files changed (1) hide show
  1. precious3_gpt_multi_modal.py +6 -2
precious3_gpt_multi_modal.py CHANGED
@@ -25,8 +25,12 @@ log = logging.getLogger(__name__)
25
 
26
 
27
  class CustomTokenizer(PreTrainedTokenizerFast):
28
- def __init__(self, vocab_file, tokenizer_file=None, **kwargs):
29
- super().__init__(vocab_file=vocab_file, tokenizer_file=tokenizer_file, **kwargs)
 
 
 
 
30
 
31
 
32
  class Custom_MptModel(MPTModel): # MptModel
 
25
 
26
 
27
  class CustomTokenizer(PreTrainedTokenizerFast):
28
+ def __init__(self, **kwargs):
29
+ super().__init__( tokenizer_file="../tokenizer.json",
30
+ unk_token="[UNK]",
31
+ pad_token="[PAD]",
32
+ eos_token="[EOS]",
33
+ bos_token="[BOS]", **kwargs)
34
 
35
 
36
  class Custom_MptModel(MPTModel): # MptModel