Package com.yahoo.embedding

Class ColBertEmbedderConfig.Builder

java.lang.Object

com.yahoo.embedding.ColBertEmbedderConfig.Builder

All Implemented Interfaces:

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

Enclosing class:

ColBertEmbedderConfig

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

Constructor Summary

Constructors

Constructor	Description
Builder()
Builder(ColBertEmbedderConfig config)

Method Summary

Modifier and Type	Method	Description
ColBertEmbedderConfig	build()
final boolean	dispatchGetConfig(com.yahoo.config.ConfigInstance.Producer producer)
ColBertEmbedderConfig.Builder	documentTokenId(int __value)
final boolean	getApplyOnRestart()
final String	getDefMd5()
final String	getDefName()
final String	getDefNamespace()
ColBertEmbedderConfig.Builder	maxDocumentTokens(int __value)
ColBertEmbedderConfig.Builder	maxQueryTokens(int __value)
ColBertEmbedderConfig.Builder	queryTokenId(int __value)
final void	setApplyOnRestart(boolean applyOnRestart)
ColBertEmbedderConfig.Builder	tokenizerPath(com.yahoo.config.ModelReference __value)
ColBertEmbedderConfig.Builder	transformerAttentionMask(String __value)
ColBertEmbedderConfig.Builder	transformerEndSequenceToken(int __value)
ColBertEmbedderConfig.Builder	transformerExecutionMode(ColBertEmbedderConfig.TransformerExecutionMode.Enum __value)
ColBertEmbedderConfig.Builder	transformerGpuDevice(int __value)
ColBertEmbedderConfig.Builder	transformerInputIds(String __value)
ColBertEmbedderConfig.Builder	transformerInterOpThreads(int __value)
ColBertEmbedderConfig.Builder	transformerIntraOpThreads(int __value)
ColBertEmbedderConfig.Builder	transformerMaskToken(int __value)
ColBertEmbedderConfig.Builder	transformerMaxTokens(int __value)
ColBertEmbedderConfig.Builder	transformerModel(com.yahoo.config.ModelReference __value)
ColBertEmbedderConfig.Builder	transformerOutput(String __value)
ColBertEmbedderConfig.Builder	transformerPadToken(int __value)
ColBertEmbedderConfig.Builder	transformerStartSequenceToken(int __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(ColBertEmbedderConfig config)

Method Details

tokenizerPath

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

transformerModel

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

maxQueryTokens

public ColBertEmbedderConfig.Builder maxQueryTokens(int __value)

maxDocumentTokens

public ColBertEmbedderConfig.Builder maxDocumentTokens(int __value)

transformerMaxTokens

public ColBertEmbedderConfig.Builder transformerMaxTokens(int __value)

transformerInputIds

public ColBertEmbedderConfig.Builder transformerInputIds(String __value)

transformerAttentionMask

public ColBertEmbedderConfig.Builder transformerAttentionMask(String __value)

transformerStartSequenceToken

public ColBertEmbedderConfig.Builder transformerStartSequenceToken(int __value)

transformerEndSequenceToken

public ColBertEmbedderConfig.Builder transformerEndSequenceToken(int __value)

transformerMaskToken

public ColBertEmbedderConfig.Builder transformerMaskToken(int __value)

transformerPadToken

public ColBertEmbedderConfig.Builder transformerPadToken(int __value)

queryTokenId

public ColBertEmbedderConfig.Builder queryTokenId(int __value)

documentTokenId

public ColBertEmbedderConfig.Builder documentTokenId(int __value)

transformerOutput

public ColBertEmbedderConfig.Builder transformerOutput(String __value)

transformerExecutionMode

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

transformerInterOpThreads

public ColBertEmbedderConfig.Builder transformerInterOpThreads(int __value)

transformerIntraOpThreads

public ColBertEmbedderConfig.Builder transformerIntraOpThreads(int __value)

transformerGpuDevice

public ColBertEmbedderConfig.Builder transformerGpuDevice(int __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 ColBertEmbedderConfig build()