mirror of
https://github.com/storytold/storyteller-ml.git
synced 2026-10-17 00:23:14 +00:00
Titles for check commands (did I test this? no?)
This commit is contained in:
@@ -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)
|
||||
|
||||
@@ -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)
|
||||
|
||||
Reference in New Issue
Block a user