正在显示
3 个修改的文件
包含
45 行增加
和
0 行删除
| @@ -13,6 +13,7 @@ java_files += FeatureConfig.java | @@ -13,6 +13,7 @@ java_files += FeatureConfig.java | ||
| 13 | java_files += OnlineLMConfig.java | 13 | java_files += OnlineLMConfig.java |
| 14 | java_files += OnlineParaformerModelConfig.java | 14 | java_files += OnlineParaformerModelConfig.java |
| 15 | java_files += OnlineZipformer2CtcModelConfig.java | 15 | java_files += OnlineZipformer2CtcModelConfig.java |
| 16 | +java_files += OnlineNeMoCtcModelConfig.java | ||
| 16 | java_files += OnlineTransducerModelConfig.java | 17 | java_files += OnlineTransducerModelConfig.java |
| 17 | java_files += OnlineModelConfig.java | 18 | java_files += OnlineModelConfig.java |
| 18 | java_files += OnlineCtcFstDecoderConfig.java | 19 | java_files += OnlineCtcFstDecoderConfig.java |
| @@ -7,6 +7,7 @@ public class OnlineModelConfig { | @@ -7,6 +7,7 @@ public class OnlineModelConfig { | ||
| 7 | private final OnlineTransducerModelConfig transducer; | 7 | private final OnlineTransducerModelConfig transducer; |
| 8 | private final OnlineParaformerModelConfig paraformer; | 8 | private final OnlineParaformerModelConfig paraformer; |
| 9 | private final OnlineZipformer2CtcModelConfig zipformer2Ctc; | 9 | private final OnlineZipformer2CtcModelConfig zipformer2Ctc; |
| 10 | + private final OnlineNeMoCtcModelConfig neMoCtc; | ||
| 10 | private final String tokens; | 11 | private final String tokens; |
| 11 | private final int numThreads; | 12 | private final int numThreads; |
| 12 | private final boolean debug; | 13 | private final boolean debug; |
| @@ -17,6 +18,7 @@ public class OnlineModelConfig { | @@ -17,6 +18,7 @@ public class OnlineModelConfig { | ||
| 17 | this.transducer = builder.transducer; | 18 | this.transducer = builder.transducer; |
| 18 | this.paraformer = builder.paraformer; | 19 | this.paraformer = builder.paraformer; |
| 19 | this.zipformer2Ctc = builder.zipformer2Ctc; | 20 | this.zipformer2Ctc = builder.zipformer2Ctc; |
| 21 | + this.neMoCtc = builder.neMoCtc; | ||
| 20 | this.tokens = builder.tokens; | 22 | this.tokens = builder.tokens; |
| 21 | this.numThreads = builder.numThreads; | 23 | this.numThreads = builder.numThreads; |
| 22 | this.debug = builder.debug; | 24 | this.debug = builder.debug; |
| @@ -40,6 +42,10 @@ public class OnlineModelConfig { | @@ -40,6 +42,10 @@ public class OnlineModelConfig { | ||
| 40 | return zipformer2Ctc; | 42 | return zipformer2Ctc; |
| 41 | } | 43 | } |
| 42 | 44 | ||
| 45 | + public OnlineNeMoCtcModelConfig getNeMoCtc() { | ||
| 46 | + return neMoCtc; | ||
| 47 | + } | ||
| 48 | + | ||
| 43 | public String getTokens() { | 49 | public String getTokens() { |
| 44 | return tokens; | 50 | return tokens; |
| 45 | } | 51 | } |
| @@ -64,6 +70,7 @@ public class OnlineModelConfig { | @@ -64,6 +70,7 @@ public class OnlineModelConfig { | ||
| 64 | private OnlineParaformerModelConfig paraformer = OnlineParaformerModelConfig.builder().build(); | 70 | private OnlineParaformerModelConfig paraformer = OnlineParaformerModelConfig.builder().build(); |
| 65 | private OnlineTransducerModelConfig transducer = OnlineTransducerModelConfig.builder().build(); | 71 | private OnlineTransducerModelConfig transducer = OnlineTransducerModelConfig.builder().build(); |
| 66 | private OnlineZipformer2CtcModelConfig zipformer2Ctc = OnlineZipformer2CtcModelConfig.builder().build(); | 72 | private OnlineZipformer2CtcModelConfig zipformer2Ctc = OnlineZipformer2CtcModelConfig.builder().build(); |
| 73 | + private OnlineNeMoCtcModelConfig neMoCtc = OnlineNeMoCtcModelConfig.builder().build(); | ||
| 67 | private String tokens = ""; | 74 | private String tokens = ""; |
| 68 | private int numThreads = 1; | 75 | private int numThreads = 1; |
| 69 | private boolean debug = true; | 76 | private boolean debug = true; |
| @@ -89,6 +96,11 @@ public class OnlineModelConfig { | @@ -89,6 +96,11 @@ public class OnlineModelConfig { | ||
| 89 | return this; | 96 | return this; |
| 90 | } | 97 | } |
| 91 | 98 | ||
| 99 | + public Builder setNeMoCtc(OnlineNeMoCtcModelConfig neMoCtc) { | ||
| 100 | + this.neMoCtc = neMoCtc; | ||
| 101 | + return this; | ||
| 102 | + } | ||
| 103 | + | ||
| 92 | public Builder setTokens(String tokens) { | 104 | public Builder setTokens(String tokens) { |
| 93 | this.tokens = tokens; | 105 | this.tokens = tokens; |
| 94 | return this; | 106 | return this; |
| 1 | +// Copyright 2024 Xiaomi Corporation | ||
| 2 | +package com.k2fsa.sherpa.onnx; | ||
| 3 | + | ||
| 4 | +public class OnlineNeMoCtcModelConfig { | ||
| 5 | + private final String model; | ||
| 6 | + | ||
| 7 | + private OnlineNeMoCtcModelConfig(Builder builder) { | ||
| 8 | + this.model = builder.model; | ||
| 9 | + } | ||
| 10 | + | ||
| 11 | + public static Builder builder() { | ||
| 12 | + return new Builder(); | ||
| 13 | + } | ||
| 14 | + | ||
| 15 | + public String getModel() { | ||
| 16 | + return model; | ||
| 17 | + } | ||
| 18 | + | ||
| 19 | + public static class Builder { | ||
| 20 | + private String model = ""; | ||
| 21 | + | ||
| 22 | + public OnlineNeMoCtcModelConfig build() { | ||
| 23 | + return new OnlineNeMoCtcModelConfig(this); | ||
| 24 | + } | ||
| 25 | + | ||
| 26 | + public Builder setModel(String model) { | ||
| 27 | + this.model = model; | ||
| 28 | + return this; | ||
| 29 | + } | ||
| 30 | + } | ||
| 31 | + | ||
| 32 | +} |
-
请 注册 或 登录 后发表评论