process

1 files changed, 4 insertions, 3 deletions

diff --git a/scripts/process.py b/scripts/process.py
index 8f5c2f0..3a2b788 100644
--- a/scripts/process.py
+++ b/scripts/process.py
@@ -10,7 +10,8 @@ import numpy
 
 from become_yukarin import AcousticConverter
 from become_yukarin import SuperResolution
-from become_yukarin.config.config import create_from_json as create_config
+from become_yukarin.config.config import create_from_json as create_conv_config
+from become_yukarin.config.sr_config import create_from_json as create_sr_config
 
 parser = argparse.ArgumentParser()
 parser.add_argument('-cmd', '--conv_model_directory', type=Path, default=Path('/mnt/dwango/hiroshiba/become-yukarin/'))
@@ -33,7 +34,7 @@ def extract_number(f):
     return int(s[-1]) if s else -1
 
 def load_acoustic_converter():
-  conv_config = create_config(conv_model / 'config.json')
+  conv_config = create_conv_config(conv_model / 'config.json')
   if it is not None:
       conv_model_path = conv_model / 'predictor_{}.npz'.format(it)
   else:
@@ -43,7 +44,7 @@ def load_acoustic_converter():
   return AcousticConverter(conv_config, conv_model_path, gpu=gpu)
 
 def load_super_resolution():
-  sr_config = create_config(sr_model / 'config.json')
+  sr_config = create_sr_config(sr_model / 'config.json')
   sr_model_paths = sr_model.glob('predictor*.npz')
   sr_model_path = list(sorted(sr_model_paths, key=extract_number))[-1]
   print(sr_model_path)