fix code style

pull/2679/head
zxcd 3 years ago
parent 445cf53f25
commit 1ae37b5e59

@ -300,7 +300,6 @@ class Tokenizer:
@lru_cache(maxsize=None)
def build_tokenizer(resource_path: str, name: str="gpt2"):
os.environ["TOKENIZERS_PARALLELISM"] = "false"
#path = os.path.join(os.path.dirname(__file__), "assets", name)
path = os.path.join(resource_path, "assets", name)
tokenizer = GPTTokenizer.from_pretrained(path)

@ -1422,7 +1422,7 @@ def hann_window(n_fft: int=N_FFT):
@lru_cache(maxsize=None)
def mel_filters(resource_path: str=None, n_mels: int=N_MELS) -> paddle.Tensor:
def mel_filters(resource_path: str, n_mels: int=N_MELS) -> paddle.Tensor:
"""
load the mel filterbank matrix for projecting STFT into a Mel spectrogram.
Allows decoupling librosa dependency; saved using:
@ -1468,7 +1468,7 @@ def log_mel_spectrogram(audio: Union[str, np.ndarray, paddle.Tensor],
magnitudes = stft[:, :-1].abs()**2
filters = mel_filters(resource_path=resource_path, n_mels=n_mels)
filters = mel_filters(resource_path, n_mels)
mel_spec = filters @ magnitudes
mel_spec = paddle.to_tensor(mel_spec.numpy().tolist())

Loading…
Cancel
Save