From 9dab23564719aac8531914574511d4b92f3d8ec2 Mon Sep 17 00:00:00 2001 From: jason-on-salt-a40 Date: Tue, 16 Apr 2024 08:55:35 -0700 Subject: [PATCH] better hf integration --- inference_speech_editing.ipynb | 4 ++-- inference_tts.ipynb | 4 ++-- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/inference_speech_editing.ipynb b/inference_speech_editing.ipynb index 4502966..022e3ba 100644 --- a/inference_speech_editing.ipynb +++ b/inference_speech_editing.ipynb @@ -203,8 +203,8 @@ "voicecraft_name=\"giga330M.pth\" # or gigaHalfLibri330M_TTSEnhanced_max16s.pth, giga830M.pth\n", "\n", "# the new way of loading the model, with huggingface, recommended\n", - "from models.voicecraft import VoiceCraftHF\n", - "model = VoiceCraftHF.from_pretrained(f\"pyp1/VoiceCraft_{voicecraft_name.replace('.pth', '')}\")\n", + "from models import voicecraft\n", + "model = voicecraft.VoiceCraft.from_pretrained(f\"pyp1/VoiceCraft_{voicecraft_name.replace('.pth', '')}\")\n", "phn2num = model.args.phn2num\n", "config = vars(model.args)\n", "model.to(device)\n", diff --git a/inference_tts.ipynb b/inference_tts.ipynb index f9a1862..e9712ca 100644 --- a/inference_tts.ipynb +++ b/inference_tts.ipynb @@ -74,8 +74,8 @@ "voicecraft_name=\"giga330M.pth\" # or gigaHalfLibri330M_TTSEnhanced_max16s.pth, giga830M.pth\n", "\n", "# the new way of loading the model, with huggingface, recommended\n", - "from models.voicecraft import VoiceCraftHF\n", - "model = VoiceCraftHF.from_pretrained(f\"pyp1/VoiceCraft_{voicecraft_name.replace('.pth', '')}\")\n", + "from models import voicecraft\n", + "model = voicecraft.VoiceCraft.from_pretrained(f\"pyp1/VoiceCraft_{voicecraft_name.replace('.pth', '')}\")\n", "phn2num = model.args.phn2num\n", "config = vars(model.args)\n", "model.to(device)\n",