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

Fix CI tests. (#907)

Browse Source
This commit is contained in:
Fangjun Kuang
2024-05-23 14:49:37 +08:00
committed by GitHub
parent 49ee458bfb
commit 2db777587e
6 changed files with 49 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
sherpa-onnx/java-api/src/com/k2fsa/sherpa/onnx/OfflineModelConfig.java
View File

@@ -13,6 +13,8 @@ public class OfflineModelConfig {
private final String provider;
private final String modelType;
private final String modelingUnit;
private final String bpeVocab;
private OfflineModelConfig(Builder builder) {
this.transducer = builder.transducer;
@@ -24,6 +26,8 @@ public class OfflineModelConfig {
this.debug = builder.debug;
this.provider = builder.provider;
this.modelType = builder.modelType;
this.modelingUnit = builder.modelingUnit;
this.bpeVocab = builder.bpeVocab;
}
public static Builder builder() {
@@ -62,6 +66,13 @@ public class OfflineModelConfig {
return modelType;
}
public String getModelingUnit() {
return modelingUnit;
}
public String getBpeVocab() {
return bpeVocab;
}
public static class Builder {
private OfflineParaformerModelConfig paraformer = OfflineParaformerModelConfig.builder().build();
@@ -73,6 +84,8 @@ public class OfflineModelConfig {
private boolean debug = true;
private String provider = "cpu";
private String modelType = "";
private String modelingUnit = "cjkchar";
private String bpeVocab = "";
public OfflineModelConfig build() {
return new OfflineModelConfig(this);
@@ -122,5 +135,13 @@ public class OfflineModelConfig {
this.modelType = modelType;
return this;
}
public void setModelingUnit(String modelingUnit) {
this.modelingUnit = modelingUnit;
}
public void setBpeVocab(String bpeVocab) {
this.bpeVocab = bpeVocab;
}
}
}

22
sherpa-onnx/java-api/src/com/k2fsa/sherpa/onnx/OnlineModelConfig.java
View File

@@ -13,6 +13,8 @@ public class OnlineModelConfig {
private final boolean debug;
private final String provider;
private final String modelType;
private final String modelingUnit;
private final String bpeVocab;
private OnlineModelConfig(Builder builder) {
this.transducer = builder.transducer;
@@ -24,6 +26,8 @@ public class OnlineModelConfig {
this.debug = builder.debug;
this.provider = builder.provider;
this.modelType = builder.modelType;
this.modelingUnit = builder.modelingUnit;
this.bpeVocab = builder.bpeVocab;
}
public static Builder builder() {
@@ -66,6 +70,14 @@ public class OnlineModelConfig {
return modelType;
}
public String getModelingUnit() {
return modelingUnit;
}
public String getBpeVocab() {
return bpeVocab;
}
public static class Builder {
private OnlineParaformerModelConfig paraformer = OnlineParaformerModelConfig.builder().build();
private OnlineTransducerModelConfig transducer = OnlineTransducerModelConfig.builder().build();
@@ -76,6 +88,8 @@ public class OnlineModelConfig {
private boolean debug = true;
private String provider = "cpu";
private String modelType = "";
private String modelingUnit = "cjkchar";
private String bpeVocab = "";
public OnlineModelConfig build() {
return new OnlineModelConfig(this);
@@ -125,5 +139,13 @@ public class OnlineModelConfig {
this.modelType = modelType;
return this;
}
public void setModelingUnit(String modelingUnit) {
this.modelingUnit = modelingUnit;
}
public void setBpeVocab(String bpeVocab) {
this.bpeVocab = bpeVocab;
}
}
}