Add Kotlin and Java API for FireRedAsr AED model (#1870)
This commit is contained in:
@@ -0,0 +1,42 @@
|
||||
package com.k2fsa.sherpa.onnx;
|
||||
|
||||
public class OfflineFireRedAsrModelConfig {
|
||||
private final String encoder;
|
||||
private final String decoder;
|
||||
|
||||
private OfflineFireRedAsrModelConfig(Builder builder) {
|
||||
this.encoder = builder.encoder;
|
||||
this.decoder = builder.decoder;
|
||||
}
|
||||
|
||||
public static Builder builder() {
|
||||
return new Builder();
|
||||
}
|
||||
|
||||
public String getEncoder() {
|
||||
return encoder;
|
||||
}
|
||||
|
||||
public String getDecoder() {
|
||||
return decoder;
|
||||
}
|
||||
|
||||
public static class Builder {
|
||||
private String encoder = "";
|
||||
private String decoder = "";
|
||||
|
||||
public OfflineFireRedAsrModelConfig build() {
|
||||
return new OfflineFireRedAsrModelConfig(this);
|
||||
}
|
||||
|
||||
public Builder setEncoder(String encoder) {
|
||||
this.encoder = encoder;
|
||||
return this;
|
||||
}
|
||||
|
||||
public Builder setDecoder(String decoder) {
|
||||
this.decoder = decoder;
|
||||
return this;
|
||||
}
|
||||
}
|
||||
}
|
||||
@@ -6,6 +6,7 @@ public class OfflineModelConfig {
|
||||
private final OfflineTransducerModelConfig transducer;
|
||||
private final OfflineParaformerModelConfig paraformer;
|
||||
private final OfflineWhisperModelConfig whisper;
|
||||
private final OfflineFireRedAsrModelConfig fireRedAsr;
|
||||
private final OfflineMoonshineModelConfig moonshine;
|
||||
private final OfflineNemoEncDecCtcModelConfig nemo;
|
||||
private final OfflineSenseVoiceModelConfig senseVoice;
|
||||
@@ -23,6 +24,7 @@ public class OfflineModelConfig {
|
||||
this.transducer = builder.transducer;
|
||||
this.paraformer = builder.paraformer;
|
||||
this.whisper = builder.whisper;
|
||||
this.fireRedAsr = builder.fireRedAsr;
|
||||
this.moonshine = builder.moonshine;
|
||||
this.nemo = builder.nemo;
|
||||
this.senseVoice = builder.senseVoice;
|
||||
@@ -96,6 +98,7 @@ public class OfflineModelConfig {
|
||||
private OfflineParaformerModelConfig paraformer = OfflineParaformerModelConfig.builder().build();
|
||||
private OfflineTransducerModelConfig transducer = OfflineTransducerModelConfig.builder().build();
|
||||
private OfflineWhisperModelConfig whisper = OfflineWhisperModelConfig.builder().build();
|
||||
private OfflineFireRedAsrModelConfig fireRedAsr = OfflineFireRedAsrModelConfig.builder().build();
|
||||
private OfflineMoonshineModelConfig moonshine = OfflineMoonshineModelConfig.builder().build();
|
||||
private OfflineNemoEncDecCtcModelConfig nemo = OfflineNemoEncDecCtcModelConfig.builder().build();
|
||||
private OfflineSenseVoiceModelConfig senseVoice = OfflineSenseVoiceModelConfig.builder().build();
|
||||
@@ -137,6 +140,11 @@ public class OfflineModelConfig {
|
||||
return this;
|
||||
}
|
||||
|
||||
public Builder setFireRedAsr(OfflineFireRedAsrModelConfig fireRedAsr) {
|
||||
this.fireRedAsr = fireRedAsr;
|
||||
return this;
|
||||
}
|
||||
|
||||
public Builder setSenseVoice(OfflineSenseVoiceModelConfig senseVoice) {
|
||||
this.senseVoice = senseVoice;
|
||||
return this;
|
||||
|
||||
Reference in New Issue
Block a user