Package ai.djl.huggingface.translator

Class ZeroShotImageClassificationTranslator.Builder

java.lang.Object

ai.djl.modality.cv.translator.BaseImageTranslator.BaseBuilder<ZeroShotImageClassificationTranslator.Builder>

ai.djl.huggingface.translator.ZeroShotImageClassificationTranslator.Builder

Enclosing class:

ZeroShotImageClassificationTranslator

public static final class ZeroShotImageClassificationTranslator.Builder
extends ai.djl.modality.cv.translator.BaseImageTranslator.BaseBuilder<ZeroShotImageClassificationTranslator.Builder>

The builder for zero-shot classification translator.

Field Summary

Fields inherited from class ai.djl.modality.cv.translator.BaseImageTranslator.BaseBuilder

batchifier, flag, height, pipeline, width

Method Summary

Modifier and Type	Method	Description
ZeroShotImageClassificationTranslator	build()	Builds the translator.
void	configure(Map<String,?> arguments)	Configures the builder with the model arguments.
ZeroShotImageClassificationTranslator.Builder	optInt32(boolean int32)	Sets if use int32 datatype for the Translator.
protected ZeroShotImageClassificationTranslator.Builder	self()

Methods inherited from class ai.djl.modality.cv.translator.BaseImageTranslator.BaseBuilder

addTransform, configPostProcess, configPreProcess, optBatchifier, optFlag, setImageSize, setPipeline, validate

Methods inherited from class java.lang.Object

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

Method Details

self

protected ZeroShotImageClassificationTranslator.Builder self()

Specified by:

self in class ai.djl.modality.cv.translator.BaseImageTranslator.BaseBuilder<ZeroShotImageClassificationTranslator.Builder>

optInt32

public ZeroShotImageClassificationTranslator.Builder optInt32(boolean int32)

Sets if use int32 datatype for the Translator.

Parameters:

int32 - 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 ZeroShotImageClassificationTranslator build()
                                            throws IOException

Builds the translator.

Returns:

the new translator

Throws:

IOException - if I/O error occurs