diff --git a/tts/vocodes_model_check_hifigan.py b/tts/vocodes_model_check_hifigan.py index 9bbce97..77fc0ee 100755 --- a/tts/vocodes_model_check_hifigan.py +++ b/tts/vocodes_model_check_hifigan.py @@ -17,6 +17,10 @@ from hifigan.env import AttrDict from hifigan.models import Generator from hifigan.denoiser import Denoiser as HifiDenoiser +from utils import print_title + +print_title('Check Hifigan Model (Tacotron)') + parser = argparse.ArgumentParser(description='Check that the HiFi-GAN model is valid') parser.add_argument('--checkpoint_path', type=str, help='path the hifigan model', required=True) diff --git a/voice_conversion/softvc/model_check_hifigan.py b/voice_conversion/softvc/model_check_hifigan.py index 403a854..4d4ff8b 100755 --- a/voice_conversion/softvc/model_check_hifigan.py +++ b/voice_conversion/softvc/model_check_hifigan.py @@ -19,6 +19,10 @@ sys.path.append("./hifigan/") from hifigan.generator import HifiganGenerator from hifigan.generator import consume_prefix_in_state_dict_if_present +from utils import print_title + +print_title('Check Hifigan Model (SoftVC)') + from gpu_utils import print_gpu_info print_gpu_info() @@ -69,4 +73,3 @@ metadata = { with open(args.output_metadata_filename, 'w') as json_file: json.dump(metadata, json_file) - \ No newline at end of file