Update TTS/utils/io.py

TTS/utils/io.py

@@ -32,20 +32,28 @@ def _ensure_torch_safe_globals():
         return
 
     safe_types = []
-    try:
+
+    try:  # XTTS config
         from TTS.tts.configs.xtts_config import XttsConfig
 
         safe_types.append(XttsConfig)
     except ImportError:
         pass
 
-    try:
+    try:  # XTTS model args
         from TTS.tts.models.xtts import XttsArgs, XttsAudioConfig
 
         safe_types.extend([XttsArgs, XttsAudioConfig])
     except ImportError:
         pass
 
+    try:  # Shared dataset configs referenced in checkpoints
+        from TTS.config.shared_configs import BaseDatasetConfig
+
+        safe_types.append(BaseDatasetConfig)
+    except ImportError:
+        pass
+
     if safe_types:
         add_safe_globals(safe_types)