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)
|