Package ai.djl.pytorch.zoo.nlp.textgeneration

Class PtGptTranslatorFactory

java.lang.Object

ai.djl.pytorch.zoo.nlp.textgeneration.PtGptTranslatorFactory

All Implemented Interfaces:

ai.djl.translate.TranslatorFactory

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

An TranslatorFactory that creates a PtGptTranslator instance.

Constructor Summary

Constructors

Constructor	Description
PtGptTranslatorFactory()

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

PtGptTranslatorFactory

public PtGptTranslatorFactory()

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