Skip to content

Commit

Permalink
Merge pull request #4 from eginhard/fix-is-multilingual
Browse files Browse the repository at this point in the history
fix(api): handle missing attribute in is_multi_lingual
  • Loading branch information
eginhard committed Mar 7, 2024
2 parents f24f7c1 + 017c84d commit 7f83dea
Show file tree
Hide file tree
Showing 2 changed files with 5 additions and 4 deletions.
2 changes: 1 addition & 1 deletion TTS/api.py
Original file line number Diff line number Diff line change
Expand Up @@ -97,7 +97,7 @@ def is_multi_lingual(self):
isinstance(self.model_name, str)
and "xtts" in self.model_name
or self.config
and ("xtts" in self.config.model or len(self.config.languages) > 1)
and ("xtts" in self.config.model or "languages" in self.config and len(self.config.languages) > 1)
):
return True
if hasattr(self.synthesizer.tts_model, "language_manager") and self.synthesizer.tts_model.language_manager:
Expand Down
7 changes: 4 additions & 3 deletions TTS/tts/datasets/dataset.py
Original file line number Diff line number Diff line change
Expand Up @@ -4,6 +4,7 @@
import random
from typing import Dict, List, Union

import mutagen
import numpy as np
import torch
import tqdm
Expand All @@ -13,8 +14,6 @@
from TTS.utils.audio import AudioProcessor
from TTS.utils.audio.numpy_transforms import compute_energy as calculate_energy

import mutagen

# to prevent too many open files error as suggested here
# https://github.com/pytorch/pytorch/issues/11201#issuecomment-421146936
torch.multiprocessing.set_sharing_strategy("file_system")
Expand Down Expand Up @@ -47,7 +46,9 @@ def string2filename(string):
def get_audio_size(audiopath):
extension = audiopath.rpartition(".")[-1].lower()
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
return int(audio_info.length * audio_info.sample_rate)
Expand Down

0 comments on commit 7f83dea

Please sign in to comment.