cn.smartjavaai.translation.model

类 NllbModel

java.lang.Object

cn.smartjavaai.translation.model.NllbModel

所有已实现的接口:

TranslationModel, AutoCloseable

public class NllbModel
extends Object
implements TranslationModel

机器翻译通用检测模型

作者:

lwx

构造器概要

构造器

构造器和说明
NllbModel()

方法概要

限定符和类型	方法和说明
void	close()
org.apache.commons.pool2.impl.GenericObjectPool<ai.djl.inference.Predictor<?,?>>	getDecode2PredictorPool()
org.apache.commons.pool2.impl.GenericObjectPool<ai.djl.inference.Predictor<?,?>>	getDecodePredictorPool()
org.apache.commons.pool2.impl.GenericObjectPool<ai.djl.inference.Predictor<?,?>>	getEncodePredictorPool()
ai.djl.ndarray.NDArray	greedyStepGen(NllbSearchConfig config, ai.djl.ndarray.NDArray pastOutputIds, ai.djl.ndarray.NDArray next_token_scores, ai.djl.ndarray.NDManager manager)
void	loadModel(TranslationModelConfig config) 加载模型
R<String>	translate(TranslateParam translateParam) 机器翻译

从类继承的方法 java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

构造器详细资料

NllbModel

public NllbModel()

方法详细资料

loadModel

public void loadModel(TranslationModelConfig config)

从接口复制的说明: TranslationModel

加载模型

指定者:

loadModel 在接口中 TranslationModel

translate

public R<String> translate(TranslateParam translateParam)

从接口复制的说明: TranslationModel

机器翻译

指定者:

translate 在接口中 TranslationModel

参数:

translateParam - 翻译参数

返回:

greedyStepGen

public ai.djl.ndarray.NDArray greedyStepGen(NllbSearchConfig config,
                                            ai.djl.ndarray.NDArray pastOutputIds,
                                            ai.djl.ndarray.NDArray next_token_scores,
                                            ai.djl.ndarray.NDManager manager)

getEncodePredictorPool

public org.apache.commons.pool2.impl.GenericObjectPool<ai.djl.inference.Predictor<?,?>> getEncodePredictorPool()

getDecodePredictorPool

public org.apache.commons.pool2.impl.GenericObjectPool<ai.djl.inference.Predictor<?,?>> getDecodePredictorPool()

getDecode2PredictorPool

public org.apache.commons.pool2.impl.GenericObjectPool<ai.djl.inference.Predictor<?,?>> getDecode2PredictorPool()

close

public void close()
           throws Exception

指定者:

close 在接口中 AutoCloseable

抛出:

Exception