Update modules/ChatTTS/ChatTTS/core.py

modules/ChatTTS/ChatTTS/core.py

@@ -23,6 +23,7 @@ from .utils.io_utils import get_latest_modified_file
 
 logging.basicConfig(level=logging.INFO)
 
+torch.serialization.add_safe_globals([BertTokenizerFast])
 
 class Chat:
     def __init__(
@@ -173,7 +174,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})
+            torch.serialization.add_safe_globals([BertTokenizerFast])
             tokenizer = torch.load(tokenizer_path, map_location=map_location)
             tokenizer.padding_side = "left"
             self.pretrain_models["tokenizer"] = tokenizer