Fixing import path due to updated naming

Addressing the following open issue: serp-ai#35
dgodevais · May 13, 2024 · 550259b · 550259b
1 parent 412ba2c
commit 550259b
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/clone_voice.ipynb b/clone_voice.ipynb
@@ -23,7 +23,7 @@
    "outputs": [],
    "source": [
     "# From https://github.com/gitmylo/bark-voice-cloning-HuBERT-quantizer\n",
-    "from hubert.hubert_manager import HuBERTManager\n",
+    "from bark_hubert_quantizer.hubert_manager import HuBERTManager\n",
     "hubert_manager = HuBERTManager()\n",
     "hubert_manager.make_sure_hubert_installed()\n",
     "hubert_manager.make_sure_tokenizer_installed()"
@@ -37,8 +37,8 @@
    "source": [
     "# From https://github.com/gitmylo/bark-voice-cloning-HuBERT-quantizer \n",
     "# Load HuBERT for semantic tokens\n",
-    "from hubert.pre_kmeans_hubert import CustomHubert\n",
-    "from hubert.customtokenizer import CustomTokenizer\n",
+    "from bark_hubert_quantizer.pre_kmeans_hubert import CustomHubert\n",
+    "from bark_hubert_quantizer.customtokenizer import CustomTokenizer\n",
     "\n",
     "# Load the HuBERT model\n",
     "hubert_model = CustomHubert(checkpoint_path='data/models/hubert/hubert.pt').to(device)\n",