@@ -13,6 +13,8 @@ public class OnlineModelConfig {
13
13
private final boolean debug ;
14
14
private final String provider ;
15
15
private final String modelType ;
16
+ private final String modelingUnit ;
17
+ private final String bpeVocab ;
16
18
17
19
private OnlineModelConfig (Builder builder ) {
18
20
this .transducer = builder .transducer ;
@@ -24,6 +26,8 @@ private OnlineModelConfig(Builder builder) {
24
26
this .debug = builder .debug ;
25
27
this .provider = builder .provider ;
26
28
this .modelType = builder .modelType ;
29
+ this .modelingUnit = builder .modelingUnit ;
30
+ this .bpeVocab = builder .bpeVocab ;
27
31
}
28
32
29
33
public static Builder builder () {
@@ -66,6 +70,14 @@ public String getModelType() {
66
70
return modelType ;
67
71
}
68
72
73
+ public String getModelingUnit () {
74
+ return modelingUnit ;
75
+ }
76
+
77
+ public String getBpeVocab () {
78
+ return bpeVocab ;
79
+ }
80
+
69
81
public static class Builder {
70
82
private OnlineParaformerModelConfig paraformer = OnlineParaformerModelConfig .builder ().build ();
71
83
private OnlineTransducerModelConfig transducer = OnlineTransducerModelConfig .builder ().build ();
@@ -76,6 +88,8 @@ public static class Builder {
76
88
private boolean debug = true ;
77
89
private String provider = "cpu" ;
78
90
private String modelType = "" ;
91
+ private String modelingUnit = "cjkchar" ;
92
+ private String bpeVocab = "" ;
79
93
80
94
public OnlineModelConfig build () {
81
95
return new OnlineModelConfig (this );
@@ -125,5 +139,13 @@ public Builder setModelType(String modelType) {
125
139
this .modelType = modelType ;
126
140
return this ;
127
141
}
142
+
143
+ public void setModelingUnit (String modelingUnit ) {
144
+ this .modelingUnit = modelingUnit ;
145
+ }
146
+
147
+ public void setBpeVocab (String bpeVocab ) {
148
+ this .bpeVocab = bpeVocab ;
149
+ }
128
150
}
129
151
}
0 commit comments