Zimix commited on
Commit
4a2a9ab
1 Parent(s): 24111c0

Update interaction.py

Browse files
Files changed (1) hide show
  1. interaction.py +1 -1
interaction.py CHANGED
@@ -20,7 +20,7 @@ class MindBot(object):
20
  else:
21
  self.model = SteamGenerationMixin.from_pretrained(model_path, device_map='auto',use_auth_token='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').half().eval()
22
 
23
- self.tokenizer = AutoTokenizer.from_pretrained(tokenizer_path)
24
  # sp_tokens = {'additional_special_tokens': ['<human>', '<bot>']}
25
  # self.tokenizer.add_special_tokens(sp_tokens)
26
  self.history = []
 
20
  else:
21
  self.model = SteamGenerationMixin.from_pretrained(model_path, device_map='auto',use_auth_token='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').half().eval()
22
 
23
+ self.tokenizer = AutoTokenizer.from_pretrained(tokenizer_path,use_auth_token='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')
24
  # sp_tokens = {'additional_special_tokens': ['<human>', '<bot>']}
25
  # self.tokenizer.add_special_tokens(sp_tokens)
26
  self.history = []