Inspire/ai-voice-cloning
1
0
Fork 0
forked from mrq/ai-voice-cloning
Code Pull Requests Activity

wrap checking for metadata when adding a voice in case it throws an error

Browse Source
This commit is contained in:
mrq 2023-02-21 17:35:30 +00:00
parent b6f7aa6264
commit ed2cf9f5ee
1 changed files with 24 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

34
src/utils.py
View File

@ -490,6 +490,17 @@ def stop_training():
training_process.kill() training_process.kill()
return "Training cancelled" return "Training cancelled"
def convert_to_halfp():
autoregressive_model_path = get_model_path('autoregressive.pth')
model = torch.load(autoregressive_model_path)
for k in model:
if re.findall(r'\.weight$', k):
print(f"Converting: {k}")
model[k] = model[k].half()
torch.save(model, './models/tortoise/autoregressive_half.pth')
print('Converted model to half precision: ./models/tortoise/autoregressive_half.pth')
def prepare_dataset( files, outdir, language=None, progress=None ): def prepare_dataset( files, outdir, language=None, progress=None ):
unload_tts() unload_tts()
@ -961,17 +972,20 @@ def read_generate_settings(file, read_latents=True):
if isinstance(file, list) and len(file) == 1: if isinstance(file, list) and len(file) == 1:
file = file[0] file = file[0]
if file is not None: try:
if hasattr(file, 'name'): if file is not None:
file = file.name if hasattr(file, 'name'):
file = file.name
if file[-4:] == ".wav": if file[-4:] == ".wav":
metadata = music_tag.load_file(file) metadata = music_tag.load_file(file)
if 'lyrics' in metadata: if 'lyrics' in metadata:
j = json.loads(str(metadata['lyrics'])) j = json.loads(str(metadata['lyrics']))
elif file[-5:] == ".json": elif file[-5:] == ".json":
with open(file, 'r') as f: with open(file, 'r') as f:
j = json.load(f) j = json.load(f)
except Exception as e:
pass
if j is None: if j is None:
print("No metadata found in audio file to read") print("No metadata found in audio file to read")