Update modules/ChatTTS/ChatTTS/core.py

modules/ChatTTS/ChatTTS/core.py

@@ -9,7 +9,7 @@ from vocos import Vocos
 
 from transformers.models.bert.tokenization_bert_fast import BertTokenizerFast
 import torch.serialization
-from tokenizers import AddedToken
+from tokenizers import AddedToken, Tokenizer
 
 from .infer.api import infer_code, refine_text
 from .model.dvae import DVAE
@@ -175,7 +175,7 @@ class Chat:
             self.logger.log(logging.INFO, "decoder loaded.")
 
         if tokenizer_path:
-            torch.serialization.add_safe_globals([BertTokenizerFast, AddedToken])
+            torch.serialization.add_safe_globals([BertTokenizerFast, AddedToken, Tokenizer])
             tokenizer = torch.load(tokenizer_path, map_location=map_location)
             tokenizer.padding_side = "left"
             self.pretrain_models["tokenizer"] = tokenizer