diff --git a/demo_part3.ipynb b/demo_part3.ipynb index 2033310..9b2bd34 100644 --- a/demo_part3.ipynb +++ b/demo_part3.ipynb @@ -105,6 +105,8 @@ " speaker_key = speaker_key.lower().replace('_', '-')\n", " \n", " source_se = torch.load(f'checkpoints_v2/base_speakers/ses/{speaker_key}.pth', map_location=device)\n", + " if torch.backends.mps.is_available() and device == 'cpu':\n", + " torch.backends.mps.is_available = lambda: False\n", " model.tts_to_file(text, speaker_id, src_path, speed=speed)\n", " save_path = f'{output_dir}/output_v2_{speaker_key}.wav'\n", "\n",