Package com.yahoo.embedding.huggingface

Class HuggingFaceEmbedderConfig.Builder

java.lang.Object

com.yahoo.embedding.huggingface.HuggingFaceEmbedderConfig.Builder

All Implemented Interfaces:

com.yahoo.config.ConfigBuilder, com.yahoo.config.ConfigInstance.Builder

Enclosing class:

HuggingFaceEmbedderConfig

public static final class HuggingFaceEmbedderConfig.Builder
extends Object
implements com.yahoo.config.ConfigInstance.Builder

Constructor Summary

Constructors

Constructor	Description
Builder()
Builder(HuggingFaceEmbedderConfig config)

Method Summary

Modifier and Type	Method	Description
HuggingFaceEmbedderConfig	build()
final boolean	dispatchGetConfig(com.yahoo.config.ConfigInstance.Producer producer)
final boolean	getApplyOnRestart()
final String	getDefMd5()
final String	getDefName()
final String	getDefNamespace()
HuggingFaceEmbedderConfig.Builder	normalize(boolean __value)
HuggingFaceEmbedderConfig.Builder	poolingStrategy(HuggingFaceEmbedderConfig.PoolingStrategy.Enum __value)
HuggingFaceEmbedderConfig.Builder	prependDocument(String __value)
HuggingFaceEmbedderConfig.Builder	prependQuery(String __value)
final void	setApplyOnRestart(boolean applyOnRestart)
HuggingFaceEmbedderConfig.Builder	tokenizerPath(com.yahoo.config.ModelReference __value)
HuggingFaceEmbedderConfig.Builder	transformerAttentionMask(String __value)
HuggingFaceEmbedderConfig.Builder	transformerExecutionMode(HuggingFaceEmbedderConfig.TransformerExecutionMode.Enum __value)
HuggingFaceEmbedderConfig.Builder	transformerGpuDevice(int __value)
HuggingFaceEmbedderConfig.Builder	transformerInputIds(String __value)
HuggingFaceEmbedderConfig.Builder	transformerInterOpThreads(int __value)
HuggingFaceEmbedderConfig.Builder	transformerIntraOpThreads(int __value)
HuggingFaceEmbedderConfig.Builder	transformerMaxTokens(int __value)
HuggingFaceEmbedderConfig.Builder	transformerModel(com.yahoo.config.ModelReference __value)
HuggingFaceEmbedderConfig.Builder	transformerOnnxConfigPath(Optional<com.yahoo.config.FileReference> __value)
HuggingFaceEmbedderConfig.Builder	transformerOutput(String __value)
HuggingFaceEmbedderConfig.Builder	transformerTokenTypeIds(String __value)

Methods inherited from class java.lang.Object

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

Constructor Details

Builder

public Builder()

Builder

public Builder(HuggingFaceEmbedderConfig config)

Method Details

tokenizerPath

public HuggingFaceEmbedderConfig.Builder tokenizerPath(com.yahoo.config.ModelReference __value)

transformerModel

public HuggingFaceEmbedderConfig.Builder transformerModel(com.yahoo.config.ModelReference __value)

transformerMaxTokens

public HuggingFaceEmbedderConfig.Builder transformerMaxTokens(int __value)

transformerInputIds

public HuggingFaceEmbedderConfig.Builder transformerInputIds(String __value)

transformerAttentionMask

public HuggingFaceEmbedderConfig.Builder transformerAttentionMask(String __value)

transformerTokenTypeIds

public HuggingFaceEmbedderConfig.Builder transformerTokenTypeIds(String __value)

transformerOutput

public HuggingFaceEmbedderConfig.Builder transformerOutput(String __value)

prependQuery

public HuggingFaceEmbedderConfig.Builder prependQuery(String __value)

prependDocument

public HuggingFaceEmbedderConfig.Builder prependDocument(String __value)

normalize

public HuggingFaceEmbedderConfig.Builder normalize(boolean __value)

poolingStrategy

public HuggingFaceEmbedderConfig.Builder poolingStrategy(HuggingFaceEmbedderConfig.PoolingStrategy.Enum __value)

transformerExecutionMode

public HuggingFaceEmbedderConfig.Builder transformerExecutionMode(HuggingFaceEmbedderConfig.TransformerExecutionMode.Enum __value)

transformerInterOpThreads

public HuggingFaceEmbedderConfig.Builder transformerInterOpThreads(int __value)

transformerIntraOpThreads

public HuggingFaceEmbedderConfig.Builder transformerIntraOpThreads(int __value)

transformerGpuDevice

public HuggingFaceEmbedderConfig.Builder transformerGpuDevice(int __value)

transformerOnnxConfigPath

public HuggingFaceEmbedderConfig.Builder transformerOnnxConfigPath(Optional<com.yahoo.config.FileReference> __value)

dispatchGetConfig

public final boolean dispatchGetConfig(com.yahoo.config.ConfigInstance.Producer producer)

Specified by:

dispatchGetConfig in interface com.yahoo.config.ConfigInstance.Builder

getDefMd5

public final String getDefMd5()

Specified by:

getDefMd5 in interface com.yahoo.config.ConfigInstance.Builder

getDefName

public final String getDefName()

Specified by:

getDefName in interface com.yahoo.config.ConfigInstance.Builder

getDefNamespace

public final String getDefNamespace()

Specified by:

getDefNamespace in interface com.yahoo.config.ConfigInstance.Builder

getApplyOnRestart

public final boolean getApplyOnRestart()

Specified by:

getApplyOnRestart in interface com.yahoo.config.ConfigInstance.Builder

setApplyOnRestart

public final void setApplyOnRestart(boolean applyOnRestart)

Specified by:

setApplyOnRestart in interface com.yahoo.config.ConfigInstance.Builder

build

public HuggingFaceEmbedderConfig build()