ProbablyBlue commited on
Commit
a41a7e3
·
verified ·
1 Parent(s): d673b3d

Update modules/ChatTTS/ChatTTS/core.py

Browse files
Files changed (1) hide show
  1. modules/ChatTTS/ChatTTS/core.py +1 -1
modules/ChatTTS/ChatTTS/core.py CHANGED
@@ -176,7 +176,7 @@ class Chat:
176
 
177
  if tokenizer_path:
178
  torch.serialization.add_safe_globals([BertTokenizerFast, AddedToken, Tokenizer])
179
- tokenizer = torch.load(tokenizer_path, map_location=map_location)
180
  tokenizer.padding_side = "left"
181
  self.pretrain_models["tokenizer"] = tokenizer
182
  self.logger.log(logging.INFO, "tokenizer loaded.")
 
176
 
177
  if tokenizer_path:
178
  torch.serialization.add_safe_globals([BertTokenizerFast, AddedToken, Tokenizer])
179
+ tokenizer = torch.load(tokenizer_path, map_location=map_location, weights_only=False)
180
  tokenizer.padding_side = "left"
181
  self.pretrain_models["tokenizer"] = tokenizer
182
  self.logger.log(logging.INFO, "tokenizer loaded.")