inoryQwQ commited on
Commit
f49e047
·
1 Parent(s): 6726998

fix model path

Browse files
Files changed (2) hide show
  1. .gitignore +1 -0
  2. python/melotts.py +5 -4
.gitignore ADDED
@@ -0,0 +1 @@
 
 
1
+ __pycache__
python/melotts.py CHANGED
@@ -130,15 +130,16 @@ def main():
130
 
131
  if enc_model is None:
132
  if "ZH" in language:
133
- enc_model = "../models/encoder-zh.onnx"
134
  else:
135
- enc_model = f"../models/encoder-{language.lower()}.onnx"
136
  assert os.path.exists(enc_model), f"Encoder model ({enc_model}) not exist!"
137
  if dec_model is None:
 
138
  if "ZH" in language:
139
- dec_model = "../models/decoder-zh.axmodel"
140
  else:
141
- dec_model = f"../models/decoder-{language.lower()}.axmodel"
142
  assert os.path.exists(dec_model), f"Decoder model ({dec_model}) not exist!"
143
 
144
  print(f"sentence: {sentence}")
 
130
 
131
  if enc_model is None:
132
  if "ZH" in language:
133
+ enc_model = "../encoder-onnx/encoder-zh.onnx"
134
  else:
135
+ enc_model = f"../encoder-onnx/encoder-{language.lower()}.onnx"
136
  assert os.path.exists(enc_model), f"Encoder model ({enc_model}) not exist!"
137
  if dec_model is None:
138
+ decoder_root = 'decoder-ax650'
139
  if "ZH" in language:
140
+ dec_model = f"../{decoder_root}/decoder-zh.axmodel"
141
  else:
142
+ dec_model = f"../{decoder_root}/decoder-{language.lower()}.axmodel"
143
  assert os.path.exists(dec_model), f"Decoder model ({dec_model}) not exist!"
144
 
145
  print(f"sentence: {sentence}")