Turbo Speed: Vectorized FFT F0 pitch, cuDNN autotune, Tensor Core precision, speaker RAM cache & prefetch DataLoader

train.py

@@ -680,13 +680,14 @@ def train(
         if is_main:
             print("[*] No local dataset found. Auto-downloading ViuAI_TTS_Dataset from Hugging Face...")
             try:
-                from huggingface_hub import snapshot_download
+                from huggingface_hub import snapshot_download, get_token
                 chosen_dir = "./dataset"
+                hf_tok = os.environ.get("HF_TOKEN") or get_token()
                 snapshot_download(
                     "ViuAI/ViuAI_TTS_Dataset",
                     repo_type="dataset",
                     local_dir=chosen_dir,
-                    token=os.environ.get("HF_TOKEN"),
+                    token=hf_tok,
                     max_workers=8,
                 )
                 print(f"[OK] Dataset downloaded successfully to: {chosen_dir}")