Add Kotlin and Java API for Moonshine models (#1474)
This commit is contained in:
@@ -6,6 +6,7 @@ public class OfflineModelConfig {
|
||||
private final OfflineTransducerModelConfig transducer;
|
||||
private final OfflineParaformerModelConfig paraformer;
|
||||
private final OfflineWhisperModelConfig whisper;
|
||||
private final OfflineMoonshineModelConfig moonshine;
|
||||
private final OfflineNemoEncDecCtcModelConfig nemo;
|
||||
private final OfflineSenseVoiceModelConfig senseVoice;
|
||||
private final String teleSpeech;
|
||||
@@ -22,6 +23,7 @@ public class OfflineModelConfig {
|
||||
this.transducer = builder.transducer;
|
||||
this.paraformer = builder.paraformer;
|
||||
this.whisper = builder.whisper;
|
||||
this.moonshine = builder.moonshine;
|
||||
this.nemo = builder.nemo;
|
||||
this.senseVoice = builder.senseVoice;
|
||||
this.teleSpeech = builder.teleSpeech;
|
||||
@@ -50,6 +52,10 @@ public class OfflineModelConfig {
|
||||
return whisper;
|
||||
}
|
||||
|
||||
public OfflineMoonshineModelConfig getMoonshine() {
|
||||
return moonshine;
|
||||
}
|
||||
|
||||
public OfflineSenseVoiceModelConfig getSenseVoice() {
|
||||
return senseVoice;
|
||||
}
|
||||
@@ -90,6 +96,7 @@ public class OfflineModelConfig {
|
||||
private OfflineParaformerModelConfig paraformer = OfflineParaformerModelConfig.builder().build();
|
||||
private OfflineTransducerModelConfig transducer = OfflineTransducerModelConfig.builder().build();
|
||||
private OfflineWhisperModelConfig whisper = OfflineWhisperModelConfig.builder().build();
|
||||
private OfflineMoonshineModelConfig moonshine = OfflineMoonshineModelConfig.builder().build();
|
||||
private OfflineNemoEncDecCtcModelConfig nemo = OfflineNemoEncDecCtcModelConfig.builder().build();
|
||||
private OfflineSenseVoiceModelConfig senseVoice = OfflineSenseVoiceModelConfig.builder().build();
|
||||
private String teleSpeech = "";
|
||||
@@ -135,6 +142,11 @@ public class OfflineModelConfig {
|
||||
return this;
|
||||
}
|
||||
|
||||
public Builder setMoonshine(OfflineMoonshineModelConfig moonshine) {
|
||||
this.moonshine = moonshine;
|
||||
return this;
|
||||
}
|
||||
|
||||
public Builder setTokens(String tokens) {
|
||||
this.tokens = tokens;
|
||||
return this;
|
||||
|
||||
@@ -0,0 +1,70 @@
|
||||
// Copyright 2024 Xiaomi Corporation
|
||||
|
||||
package com.k2fsa.sherpa.onnx;
|
||||
|
||||
public class OfflineMoonshineModelConfig {
|
||||
private final String preprocessor;
|
||||
private final String encoder;
|
||||
private final String uncachedDecoder;
|
||||
private final String cachedDecoder;
|
||||
|
||||
private OfflineMoonshineModelConfig(Builder builder) {
|
||||
this.preprocessor = builder.preprocessor;
|
||||
this.encoder = builder.encoder;
|
||||
this.uncachedDecoder = builder.uncachedDecoder;
|
||||
this.cachedDecoder = builder.cachedDecoder;
|
||||
}
|
||||
|
||||
public static Builder builder() {
|
||||
return new Builder();
|
||||
}
|
||||
|
||||
public String getPreprocessor() {
|
||||
return preprocessor;
|
||||
}
|
||||
|
||||
public String getEncoder() {
|
||||
return encoder;
|
||||
}
|
||||
|
||||
public String getUncachedDecoder() {
|
||||
return uncachedDecoder;
|
||||
}
|
||||
|
||||
public String getCachedDecoder() {
|
||||
return cachedDecoder;
|
||||
}
|
||||
|
||||
public static class Builder {
|
||||
private String preprocessor = "";
|
||||
private String encoder = "";
|
||||
private String uncachedDecoder = "";
|
||||
private String cachedDecoder = "";
|
||||
|
||||
public OfflineMoonshineModelConfig build() {
|
||||
return new OfflineMoonshineModelConfig(this);
|
||||
}
|
||||
|
||||
public Builder setPreprocessor(String preprocessor) {
|
||||
this.preprocessor = preprocessor;
|
||||
return this;
|
||||
}
|
||||
|
||||
public Builder setEncoder(String encoder) {
|
||||
this.encoder = encoder;
|
||||
return this;
|
||||
}
|
||||
|
||||
public Builder setUncachedDecoder(String uncachedDecoder) {
|
||||
this.uncachedDecoder = uncachedDecoder;
|
||||
return this;
|
||||
}
|
||||
|
||||
public Builder setCachedDecoder(String cachedDecoder) {
|
||||
this.cachedDecoder = cachedDecoder;
|
||||
return this;
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
}
|
||||
Reference in New Issue
Block a user