githubmirror/MockingBird

utils/modelutils.py

16 строк767 B

Vega

Add gst (#137)

12 окт 2021, 14:43

12 окт 2021, 14:432a99f0f

from pathlib import Path

def check_model_paths(encoder_path: Path, synthesizer_path: Path, vocoder_path: Path):
    # This function tests the model paths and makes sure at least one is valid.
    if encoder_path.is_file() or encoder_path.is_dir():
        return
    if synthesizer_path.is_file() or synthesizer_path.is_dir():
        return
    if vocoder_path.is_file() or vocoder_path.is_dir():
        return

    # If none of the paths exist, remind the user to download models if needed
    print("********************************************************************************")
    print("Error: Model files not found. Please download the models")
    print("********************************************************************************\n")
    quit(-1)