Skip to content

Commit 8742c63

Browse files
committed
small fixes
1 parent 57ddf3d commit 8742c63

3 files changed

Lines changed: 21 additions & 3 deletions

File tree

sherpa-onnx/csrc/offline-tts-kitten-impl.h

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -261,7 +261,7 @@ class OfflineTtsKittenImpl : public OfflineTtsImpl {
261261
std::vector<std::vector<int64_t>> batch_x;
262262

263263
int32_t batch_size = 1;
264-
batch_x.reserve(config_.max_num_sentences);
264+
batch_x.reserve(batch_size);
265265
int32_t num_batches = x_size / batch_size;
266266

267267
if (config_.model.debug) {

sherpa-onnx/csrc/offline-tts-kitten-model-config.cc

Lines changed: 18 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -35,6 +35,16 @@ bool OfflineTtsKittenModelConfig::Validate() const {
3535
return false;
3636
}
3737

38+
if (voices.empty()) {
39+
SHERPA_ONNX_LOGE("Please provide --kitten-voices");
40+
return false;
41+
}
42+
43+
if (!FileExists(voices)) {
44+
SHERPA_ONNX_LOGE("--kitten-voices: '%s' does not exist", voices.c_str());
45+
return false;
46+
}
47+
3848
if (tokens.empty()) {
3949
SHERPA_ONNX_LOGE("Please provide --kitten-tokens");
4050
return false;
@@ -78,6 +88,14 @@ bool OfflineTtsKittenModelConfig::Validate() const {
7888
return false;
7989
}
8090

91+
if (length_scale <= 0) {
92+
SHERPA_ONNX_LOGE(
93+
"Please provide a positive length_scale for --kitten-length-scale. "
94+
"Given: %.3f",
95+
length_scale);
96+
return false;
97+
}
98+
8199
return true;
82100
}
83101

sherpa-onnx/csrc/offline-tts-kitten-model.cc

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -63,7 +63,7 @@ class OfflineTtsKittenModel::Impl {
6363
if (x_shape[0] != 1) {
6464
SHERPA_ONNX_LOGE("Support only batch_size == 1. Given: %d",
6565
static_cast<int32_t>(x_shape[0]));
66-
exit(-1);
66+
SHERPA_ONNX_EXIT(-1);
6767
}
6868

6969
int32_t num_speakers = meta_data_.num_speakers;
@@ -223,7 +223,7 @@ class OfflineTtsKittenModel::Impl {
223223
OfflineTtsKittenModelMetaData meta_data_;
224224
std::vector<int32_t> style_dim_;
225225

226-
// (num_speakers, style_dim_[0], style_dim_[2])
226+
// (num_speakers, style_dim_[1])
227227
std::vector<float> styles_;
228228
};
229229

0 commit comments

Comments
 (0)