Java api update for adding modelType in config class (#228)

This commit is contained in:
zhaomingwork
2023-07-30 17:04:18 +08:00
committed by GitHub
parent daffdab52a
commit 5a549615df
4 changed files with 57 additions and 29 deletions

View File

@@ -11,15 +11,24 @@ public class OnlineTransducerModelConfig {
private final String tokens;
private final int numThreads;
private final boolean debug;
private final String provider = "cpu";
private String modelType = "";
public OnlineTransducerModelConfig(
String encoder, String decoder, String joiner, String tokens, int numThreads, boolean debug) {
String encoder,
String decoder,
String joiner,
String tokens,
int numThreads,
boolean debug,
String modelType) {
this.encoder = encoder;
this.decoder = decoder;
this.joiner = joiner;
this.tokens = tokens;
this.numThreads = numThreads;
this.debug = debug;
this.modelType = modelType;
}
public String getEncoder() {