avoid triton

openvinotoolkit · Jan 28, 2025 · f7cf62a · f7cf62a
1 parent 6132239
commit f7cf62a
Showing 1 changed file with 29 additions and 10 deletions.
diff --git a/notebooks/openvoice/openvoice.ipynb b/notebooks/openvoice/openvoice.ipynb
@@ -86,7 +86,7 @@
     "    r = requests.get(\n",
     "        url=\"https://raw.githubusercontent.com/openvinotoolkit/openvino_notebooks/latest/utils/pip_helper.py\",\n",
     "    )\n",
-    "    open(\"pip_helper.py\", \"w\").write(r.text)\n"
+    "    open(\"pip_helper.py\", \"w\").write(r.text)"
    ]
   },
   {
@@ -99,6 +99,7 @@
     "\n",
     "from cmd_helper import clone_repo\n",
     "from pip_helper import pip_install\n",
+    "import platform\n",
     "\n",
     "\n",
     "repo_dir = Path(\"OpenVoice\")\n",
@@ -107,9 +108,9 @@
     "    clone_repo(\"https://github.com/myshell-ai/OpenVoice\")\n",
     "    orig_english_path = Path(\"OpenVoice/openvoice/text/_orig_english.py\")\n",
     "    english_path = Path(\"OpenVoice/openvoice/text/english.py\")\n",
-    "    \n",
+    "\n",
     "    english_path.rename(orig_english_path)\n",
-    "    \n",
+    "\n",
     "    with orig_english_path.open(\"r\") as f:\n",
     "        data = f.read()\n",
     "        data = data.replace(\"unidecode\", \"anyascii\")\n",
@@ -120,15 +121,33 @@
     "# fix a problem with silero downloading and installing\n",
     "with Path(\"OpenVoice/openvoice/se_extractor.py\").open(\"r\") as orig_file:\n",
     "    data = orig_file.read()\n",
-    "    data = data.replace(\"method=\\\"silero\\\"\", \"method=\\\"silero:3.0\\\"\")\n",
+    "    data = data.replace('method=\"silero\"', 'method=\"silero:3.0\"')\n",
     "    with Path(\"OpenVoice/openvoice/se_extractor.py\").open(\"w\") as out_f:\n",
-    "            out_f.write(data)\n",
-    "\n",
+    "        out_f.write(data)\n",
+    "\n",
+    "\n",
+    "pip_install(\"librosa>=0.8.1\", \"pydub>=0.25.1\", \"tqdm\", \"inflect>=7.0.0\", \"pypinyin>=0.50.0\", \"openvino>=2023.3\", \"gradio>=4.15\")\n",
+    "pip_install(\n",
+    "    \"--extra-index-url\",\n",
+    "    \"https://download.pytorch.org/whl/cpu\",\n",
+    "    \"wavmark>=0.0.3\",\n",
+    "    \"faster-whisper>=0.9.0\",\n",
+    "    \"eng_to_ipa>=0.0.2\",\n",
+    "    \"cn2an>=0.5.22\",\n",
+    "    \"jieba>=0.42.1\",\n",
+    "    \"langid>=1.1.6\",\n",
+    "    \"ipywebrtc\",\n",
+    "    \"anyascii\",\n",
+    "    \"torch>=2.1\",\n",
+    "    \"nncf>=2.11.0\",\n",
+    "    \"dtw-python\",\n",
+    "    \"more-itertools\",\n",
+    "    \"tiktoken\",\n",
+    ")\n",
+    "pip_install(\"--no-deps\", \"whisper-timestamped>=1.14.2\", \"openai-whisper\")\n",
     "\n",
-    "pip_install(\"librosa>=0.8.1\", \"pydub>=0.25.1\",  \"tqdm\", \"inflect>=7.0.0\",  \"pypinyin>=0.50.0\", \"openvino>=2023.3\", \"gradio>=4.15\")\n",
-    "pip_install(\"--extra-index-url\", \"https://download.pytorch.org/whl/cpu\", \n",
-    "            \"wavmark>=0.0.3\", \"faster-whisper>=0.9.0\",  \"whisper-timestamped>=1.14.2\", \"eng_to_ipa>=0.0.2\", \n",
-    "            \"cn2an>=0.5.22\", \"jieba>=0.42.1\", \"langid>=1.1.6\", \"ipywebrtc\", \"anyascii\",  \"torch>=2.1\", \"nncf>=2.11.0\")"
+    "if platform.system() == \"Darwin\":\n",
+    "    pip_install(\"numpy<2.0\")"
    ]
   },
   {