Package com.yahoo.embedding

Class SpladeEmbedderConfig.Builder

java.lang.Object

com.yahoo.embedding.SpladeEmbedderConfig.Builder

All Implemented Interfaces:

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

Enclosing class:

SpladeEmbedderConfig

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

Constructor Summary

Constructors

Constructor	Description
Builder()
Builder(SpladeEmbedderConfig config)

Method Summary

Modifier and Type	Method	Description
SpladeEmbedderConfig	build()
final boolean	dispatchGetConfig(com.yahoo.config.ConfigInstance.Producer producer)
final boolean	getApplyOnRestart()
final String	getDefMd5()
final String	getDefName()
final String	getDefNamespace()
final void	setApplyOnRestart(boolean applyOnRestart)
SpladeEmbedderConfig.Builder	termScoreThreshold(double __value)
SpladeEmbedderConfig.Builder	tokenizerPath(com.yahoo.config.ModelReference __value)
SpladeEmbedderConfig.Builder	transformerAttentionMask(String __value)
SpladeEmbedderConfig.Builder	transformerExecutionMode(SpladeEmbedderConfig.TransformerExecutionMode.Enum __value)
SpladeEmbedderConfig.Builder	transformerGpuDevice(int __value)
SpladeEmbedderConfig.Builder	transformerInputIds(String __value)
SpladeEmbedderConfig.Builder	transformerInterOpThreads(int __value)
SpladeEmbedderConfig.Builder	transformerIntraOpThreads(int __value)
SpladeEmbedderConfig.Builder	transformerMaxTokens(int __value)
SpladeEmbedderConfig.Builder	transformerModel(com.yahoo.config.ModelReference __value)
SpladeEmbedderConfig.Builder	transformerOutput(String __value)
SpladeEmbedderConfig.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(SpladeEmbedderConfig config)

Method Details

tokenizerPath

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

transformerModel

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

transformerMaxTokens

public SpladeEmbedderConfig.Builder transformerMaxTokens(int __value)

transformerInputIds

public SpladeEmbedderConfig.Builder transformerInputIds(String __value)

transformerAttentionMask

public SpladeEmbedderConfig.Builder transformerAttentionMask(String __value)

transformerTokenTypeIds

public SpladeEmbedderConfig.Builder transformerTokenTypeIds(String __value)

transformerOutput

public SpladeEmbedderConfig.Builder transformerOutput(String __value)

termScoreThreshold

public SpladeEmbedderConfig.Builder termScoreThreshold(double __value)

transformerExecutionMode

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

transformerInterOpThreads

public SpladeEmbedderConfig.Builder transformerInterOpThreads(int __value)

transformerIntraOpThreads

public SpladeEmbedderConfig.Builder transformerIntraOpThreads(int __value)

transformerGpuDevice

public SpladeEmbedderConfig.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 SpladeEmbedderConfig build()