fix model path
Browse files- .gitignore +1 -0
- 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 = "../
|
| 134 |
else:
|
| 135 |
-
enc_model = f"../
|
| 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 = "../
|
| 140 |
else:
|
| 141 |
-
dec_model = f"../
|
| 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}")
|