Update improved_voice_clone.py

improved_voice_clone.py

@@ -11,6 +11,7 @@ import sys
 from pathlib import Path
 from scipy import signal
 
+
 # Configure logging
 logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(name)s - %(levelname)s - %(message)s')
 logger = logging.getLogger("ImprovedVoiceCloner")
@@ -21,6 +22,9 @@ class ImprovedVoiceCloner:
         Initialize the improved voice cloning system using XTTS-v2
         """
         # Set device to GPU if specified or available
+        import warnings
+        warnings.filterwarnings("ignore", category=UserWarning)
+        warnings.filterwarnings("ignore", message=".*attention_mask.*")
         if device == "cuda":
             self.use_gpu = torch.cuda.is_available()
             if not self.use_gpu: