Package ai.djl.huggingface.translator

Class QuestionAnsweringTranslator.Builder

java.lang.Object

ai.djl.huggingface.translator.QuestionAnsweringTranslator.Builder

Enclosing class:

QuestionAnsweringTranslator

public static final class QuestionAnsweringTranslator.Builder
extends Object

The builder for question answering translator.

Method Summary

Modifier and Type	Method	Description
QuestionAnsweringTranslator	build()	Builds the translator.
void	configure(Map<String,?> arguments)	Configures the builder with the model arguments.
QuestionAnsweringTranslator.Builder	optBatchifier(ai.djl.translate.Batchifier batchifier)	Sets the Batchifier for the Translator.
QuestionAnsweringTranslator.Builder	optIncludeTokenTypes(boolean includeTokenTypes)	Sets if include token types for the Translator.

Methods inherited from class java.lang.Object

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

Method Details

optIncludeTokenTypes

public QuestionAnsweringTranslator.Builder optIncludeTokenTypes(boolean includeTokenTypes)

Sets if include token types for the Translator.

Parameters:

includeTokenTypes - true to include token types

Returns:

this builder

optBatchifier

public QuestionAnsweringTranslator.Builder optBatchifier(ai.djl.translate.Batchifier batchifier)

Sets the Batchifier for the Translator.

Parameters:

batchifier - true to include token types

Returns:

this builder

configure

public void configure(Map<String,?> arguments)

Configures the builder with the model arguments.

Parameters:

arguments - the model arguments

build

public QuestionAnsweringTranslator build()
                                  throws IOException

Builds the translator.

Returns:

the new translator

Throws:

IOException - if I/O error occurs