Zimix commited on
Commit
8e684e2
·
1 Parent(s): 3a06052

Update interaction.py

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