Package ai.djl.onnxruntime.zoo.nlp.textgeneration

Class OrtGptTranslatorFactory

java.lang.Object

ai.djl.onnxruntime.zoo.nlp.textgeneration.OrtGptTranslatorFactory

All Implemented Interfaces:

ai.djl.translate.TranslatorFactory

public class OrtGptTranslatorFactory
extends Object
implements ai.djl.translate.TranslatorFactory

An TranslatorFactory that creates a OrtGptTranslator instance.

Constructor Summary

Constructors

Constructor	Description
OrtGptTranslatorFactory()

Method Summary

Modifier and Type	Method	Description
Set<ai.djl.util.Pair<Type,Type>>	getSupportedTypes()
<I, O> ai.djl.translate.Translator<I,O>	newInstance(Class<I> input, Class<O> output, ai.djl.Model model, Map<String,?> arguments)

Methods inherited from class java.lang.Object

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

Methods inherited from interface ai.djl.translate.TranslatorFactory

isSupported

Constructor Details

OrtGptTranslatorFactory

public OrtGptTranslatorFactory()

Method Details

getSupportedTypes

public Set<ai.djl.util.Pair<Type,Type>> getSupportedTypes()

Specified by:

getSupportedTypes in interface ai.djl.translate.TranslatorFactory

newInstance

public <I,
O> ai.djl.translate.Translator<I,O> newInstance(Class<I> input,
 Class<O> output,
 ai.djl.Model model,
 Map<String,?> arguments)

Specified by:

newInstance in interface ai.djl.translate.TranslatorFactory