Update modules/ChatTTS/ChatTTS/core.py

modules/ChatTTS/ChatTTS/core.py

@@ -7,6 +7,9 @@ from huggingface_hub import snapshot_download
 from omegaconf import OmegaConf
 from vocos import Vocos
 
+from transformers.models.bert.tokenization_bert_fast import BertTokenizerFast
+import torch.serialization
+
 from .infer.api import infer_code, refine_text
 from .model.dvae import DVAE
 from .model.gpt import GPT_warpper
@@ -170,6 +173,7 @@ class Chat:
             self.logger.log(logging.INFO, "decoder loaded.")
 
         if tokenizer_path:
+            torch.serialization.add_safe_globals({'transformers.models.bert.tokenization_bert_fast.BertTokenizerFast': BertTokenizerFast})
             tokenizer = torch.load(tokenizer_path, map_location=map_location)
             tokenizer.padding_side = "left"
             self.pretrain_models["tokenizer"] = tokenizer