add bark voice cloning (#2)

* Update webui.py
rsxdalv · May 21, 2023 · eaa8389 · eaa8389
1 parent 5468446
commit eaa8389
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/webui.py b/webui.py
@@ -80,6 +80,17 @@ def update_dependencies():
  run_cmd("python setup.py install")
  run_cmd("conda install -y --channel=numba llvmlite")
 
+ os.chdir(script_dir)
+ # clone if doesn't exist
+ if not os.path.exists("tts-generation-webui/models/bark_voice_cloning_hubert_quantizer"):
+ run_cmd("git clone https://github.com/rsxdalv/bark-voice-cloning-HuBERT-quantizer.git tts-generation-webui/models/bark_voice_cloning_hubert_quantizer")
+
+ os.chdir("tts-generation-webui/models/bark_voice_cloning_hubert_quantizer")
+ run_cmd("git pull")
+
+ # Installs/Updates dependencies from all requirements.txt
+ run_cmd("pip install -r requirements.txt")
+
 
 def run_model():
  os.chdir("tts-generation-webui")