Updates tts model
Browse files- models/mbmelgan.onnx +2 -2
- models/{lightspeech_processor.json → processor.json} +0 -0
- tts.py +6 -1
models/mbmelgan.onnx
CHANGED
@@ -1,3 +1,3 @@
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
-
oid sha256:
|
3 |
-
size
|
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:3514c3bd67bb793b8a823703a75bfbe7ff54b86214fe740da7db4b3eea89d22d
|
3 |
+
size 10453995
|
models/{lightspeech_processor.json → processor.json}
RENAMED
File without changes
|
tts.py
CHANGED
@@ -15,10 +15,15 @@ import logging
|
|
15 |
logging.basicConfig(level=logging.INFO)
|
16 |
logger = logging.getLogger(__name__)
|
17 |
|
|
|
|
|
|
|
|
|
|
|
18 |
# Load models
|
19 |
lightspeech = ort.InferenceSession("./models/lightspeech_quant.onnx")
|
20 |
mbmelgan = ort.InferenceSession("./models/mbmelgan.onnx")
|
21 |
-
lightspeech_processor_config = Path("./models/
|
22 |
|
23 |
with open(lightspeech_processor_config, "r") as f:
|
24 |
processor = json.load(f)
|
|
|
15 |
logging.basicConfig(level=logging.INFO)
|
16 |
logger = logging.getLogger(__name__)
|
17 |
|
18 |
+
# Load models
|
19 |
+
# lightspeech = ort.InferenceSession("./models/lightspeech-mfa-sw-v4/lightspeech_quant.onnx")
|
20 |
+
# mbmelgan = ort.InferenceSession("./models/mb-melgan-hifi-postnets-sw-v4/mbmelgan.onnx")
|
21 |
+
# lightspeech_processor_config = Path("./models/lightspeech-mfa-sw-v4/processor.json")
|
22 |
+
|
23 |
# Load models
|
24 |
lightspeech = ort.InferenceSession("./models/lightspeech_quant.onnx")
|
25 |
mbmelgan = ort.InferenceSession("./models/mbmelgan.onnx")
|
26 |
+
lightspeech_processor_config = Path("./models/processor.json")
|
27 |
|
28 |
with open(lightspeech_processor_config, "r") as f:
|
29 |
processor = json.load(f)
|