Commit ·
f33e451
1
Parent(s): f8c53d2
modified config and handler to get the runner path correct
Browse files- handler.py +5 -5
- or/fastpitch/config.json +2 -2
handler.py
CHANGED
|
@@ -115,12 +115,12 @@ class EndpointHandler():
|
|
| 115 |
print(f"cuda version>> ",{torch.cuda.get_device_capability(0)[0]})
|
| 116 |
models = {}
|
| 117 |
odia_model = Synthesizer(
|
| 118 |
-
tts_checkpoint="
|
| 119 |
-
tts_config_path="
|
| 120 |
# modify this config.json to proper model path
|
| 121 |
-
tts_speakers_file="
|
| 122 |
-
vocoder_checkpoint="
|
| 123 |
-
vocoder_config="
|
| 124 |
use_cuda = True
|
| 125 |
)
|
| 126 |
models["or"] = odia_model
|
|
|
|
| 115 |
print(f"cuda version>> ",{torch.cuda.get_device_capability(0)[0]})
|
| 116 |
models = {}
|
| 117 |
odia_model = Synthesizer(
|
| 118 |
+
tts_checkpoint="/repository/or/fastpitch/best_model.pth",
|
| 119 |
+
tts_config_path="/repository/or/fastpitch/config.json",
|
| 120 |
# modify this config.json to proper model path
|
| 121 |
+
tts_speakers_file="/repository/or/fastpitch/speakers.pth",
|
| 122 |
+
vocoder_checkpoint="/repository/or/hifigan/best_model.pth",
|
| 123 |
+
vocoder_config="/repository/or/hifigan/config.json",
|
| 124 |
use_cuda = True
|
| 125 |
)
|
| 126 |
models["or"] = odia_model
|
or/fastpitch/config.json
CHANGED
|
@@ -178,7 +178,7 @@
|
|
| 178 |
"max_duration": 75,
|
| 179 |
"num_speakers": 2,
|
| 180 |
"use_speaker_embedding": true,
|
| 181 |
-
"speakers_file": "
|
| 182 |
"use_d_vector_file": false,
|
| 183 |
"d_vector_dim": 512,
|
| 184 |
"d_vector_file": null,
|
|
@@ -191,7 +191,7 @@
|
|
| 191 |
},
|
| 192 |
"return_wav": false,
|
| 193 |
"num_speakers": 2,
|
| 194 |
-
"speakers_file": "
|
| 195 |
"use_speaker_embedding": true,
|
| 196 |
"use_d_vector_file": false,
|
| 197 |
"d_vector_file": "",
|
|
|
|
| 178 |
"max_duration": 75,
|
| 179 |
"num_speakers": 2,
|
| 180 |
"use_speaker_embedding": true,
|
| 181 |
+
"speakers_file": "/repository/or/fastpitch/speakers.pth",
|
| 182 |
"use_d_vector_file": false,
|
| 183 |
"d_vector_dim": 512,
|
| 184 |
"d_vector_file": null,
|
|
|
|
| 191 |
},
|
| 192 |
"return_wav": false,
|
| 193 |
"num_speakers": 2,
|
| 194 |
+
"speakers_file": "/repository/or/fastpitch/speakers.pth",
|
| 195 |
"use_speaker_embedding": true,
|
| 196 |
"use_d_vector_file": false,
|
| 197 |
"d_vector_file": "",
|