mrq/ai-voice-cloning
31
74
Fork 43
Code Issues 223 Pull Requests 5 Packages Projects Releases Wiki Activity

is

Browse Source
This commit is contained in:
mrq 2023-03-09 04:33:12 +00:00
parent a01eb10960
commit a182df8f4e
1 changed files with 3 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
src/utils.py
View File

@ -19,6 +19,7 @@ import gc
import subprocess import subprocess
import psutil import psutil
import yaml import yaml
import hashlib
import tqdm import tqdm
import torch import torch
@ -93,11 +94,11 @@ def generate(**kwargs):
voice_cache = {} voice_cache = {}
def fetch_voice( voice ): def fetch_voice( voice ):
print(f"Loading voice: {voice} with model {tts.autoregressive_model_hash[:8]}")
cache_key = f'{voice}:{tts.autoregressive_model_hash[:8]}' cache_key = f'{voice}:{tts.autoregressive_model_hash[:8]}'
if cache_key in voice_cache: if cache_key in voice_cache:
return voice_cache[cache_key] return voice_cache[cache_key]
print(f"Loading voice: {voice} with model {tts.autoregressive_model_hash[:8]}")
sample_voice = None sample_voice = None
if voice == "microphone": if voice == "microphone":
if parameters['mic_audio'] is None: if parameters['mic_audio'] is None:
@ -458,8 +459,6 @@ def cancel_generate():
tortoise.api.STOP_SIGNAL = True tortoise.api.STOP_SIGNAL = True
def hash_file(path, algo="md5", buffer_size=0): def hash_file(path, algo="md5", buffer_size=0):
import hashlib
hash = None hash = None
if algo == "md5": if algo == "md5":
hash = hashlib.md5() hash = hashlib.md5()
@ -1523,7 +1522,7 @@ def import_voices(files, saveAs=None, progress=None):
def get_voice_list(dir=get_voice_dir(), append_defaults=False): def get_voice_list(dir=get_voice_dir(), append_defaults=False):
defaults = [ "random", "microphone" ] defaults = [ "random", "microphone" ]
os.makedirs(dir, exist_ok=True) os.makedirs(dir, exist_ok=True)
res = sorted([d for d in os.listdir(dir) if d is not in defaults and os.path.isdir(os.path.join(dir, d)) and len(os.listdir(os.path.join(dir, d))) > 0 ]) res = sorted([d for d in os.listdir(dir) if d not in defaults and os.path.isdir(os.path.join(dir, d)) and len(os.listdir(os.path.join(dir, d))) > 0 ])
if append_defaults: if append_defaults:
res = res + defaults res = res + defaults
return res return res