Add Kotlin and Java API for FireRedAsr AED model (#1870)

This commit is contained in:
Fangjun Kuang
2025-02-17 10:50:25 +08:00
committed by GitHub
parent 316424b382
commit d148860d2c
10 changed files with 188 additions and 1 deletions

View File

@@ -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;
}
}
}

View File

@@ -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;