EngineX-Iluvatar/enginex-mr_series-sherpa-onnx
Archived
16
0
Fork 0
Code Issues Pull Requests Actions Projects Releases Wiki Activity

Add Java API for Matcha-TTS models. (#1673)

Browse Source
This commit is contained in:
Fangjun Kuang
2025-01-02 15:15:30 +08:00
committed by GitHub
parent f457baea42
commit a00d3b4821
11 changed files with 359 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
sherpa-onnx/java-api/src/com/k2fsa/sherpa/onnx/OfflineTtsModelConfig.java
View File

@@ -4,12 +4,14 @@ package com.k2fsa.sherpa.onnx;
public class OfflineTtsModelConfig {
private final OfflineTtsVitsModelConfig vits;
private final OfflineTtsMatchaModelConfig matcha;
private final int numThreads;
private final boolean debug;
private final String provider;
private OfflineTtsModelConfig(Builder builder) {
this.vits = builder.vits;
this.matcha = builder.matcha;
this.numThreads = builder.numThreads;
this.debug = builder.debug;
this.provider = builder.provider;
@@ -23,8 +25,13 @@ public class OfflineTtsModelConfig {
return vits;
}
public OfflineTtsMatchaModelConfig getMatcha() {
return matcha;
}
public static class Builder {
private OfflineTtsVitsModelConfig vits = OfflineTtsVitsModelConfig.builder().build();
private OfflineTtsMatchaModelConfig matcha = OfflineTtsMatchaModelConfig.builder().build();
private int numThreads = 1;
private boolean debug = true;
private String provider = "cpu";
@@ -38,6 +45,11 @@ public class OfflineTtsModelConfig {
return this;
}
public Builder setMatcha(OfflineTtsMatchaModelConfig matcha) {
this.matcha = matcha;
return this;
}
public Builder setNumThreads(int numThreads) {
this.numThreads = numThreads;
return this;