Medix-AI/coqui-tts
Medix-AI
/
coqui-tts
mirror of https://github.com/coqui-ai/TTS.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

style: make style && make lint

This commit is contained in:
Enno Hermann 2024-03-06 22:45:35 +01:00
parent 4e183c61df
commit 017c84d005
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
TTS/tts/datasets/dataset.py
View File

@ -4,6 +4,7 @@ import os
import random import random
from typing import Dict, List, Union from typing import Dict, List, Union
import mutagen
import numpy as np import numpy as np
import torch import torch
import tqdm import tqdm
@ -13,8 +14,6 @@ from TTS.tts.utils.data import prepare_data, prepare_stop_target, prepare_tensor
from TTS.utils.audio import AudioProcessor from TTS.utils.audio import AudioProcessor
from TTS.utils.audio.numpy_transforms import compute_energy as calculate_energy from TTS.utils.audio.numpy_transforms import compute_energy as calculate_energy
import mutagen
# to prevent too many open files error as suggested here # to prevent too many open files error as suggested here
# https://github.com/pytorch/pytorch/issues/11201#issuecomment-421146936 # https://github.com/pytorch/pytorch/issues/11201#issuecomment-421146936
torch.multiprocessing.set_sharing_strategy("file_system") torch.multiprocessing.set_sharing_strategy("file_system")
@ -47,7 +46,9 @@ def string2filename(string):
def get_audio_size(audiopath): def get_audio_size(audiopath):
extension = audiopath.rpartition(".")[-1].lower() extension = audiopath.rpartition(".")[-1].lower()
if extension not in {"mp3", "wav", "flac"}: if extension not in {"mp3", "wav", "flac"}:
raise RuntimeError(f"The audio format {extension} is not supported, please convert the audio files to mp3, flac, or wav format!") raise RuntimeError(
f"The audio format {extension} is not supported, please convert the audio files to mp3, flac, or wav format!"
)
audio_info = mutagen.File(audiopath).info audio_info = mutagen.File(audiopath).info
return int(audio_info.length * audio_info.sample_rate) return int(audio_info.length * audio_info.sample_rate)