Package com.yahoo.embedding

Class ColBertEmbedderConfig

java.lang.Object

com.yahoo.config.Node

com.yahoo.config.InnerNode

com.yahoo.config.ConfigInstance

com.yahoo.embedding.ColBertEmbedderConfig

public final class ColBertEmbedderConfig
extends com.yahoo.config.ConfigInstance

This class represents the root node of col-bert-embedder Copyright Vespa.ai. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.

Nested Class Summary

Nested Classes

Modifier and Type	Class	Description
static final class	ColBertEmbedderConfig.Builder
static interface	ColBertEmbedderConfig.Producer
static final class	ColBertEmbedderConfig.TransformerExecutionMode	This class represents col-bert-embedder.transformerExecutionMode Settings for ONNX model evaluation

Field Summary

Fields

Modifier and Type	Field	Description
static final String	CONFIG_DEF_MD5
static final String	CONFIG_DEF_NAME
static final String	CONFIG_DEF_NAMESPACE
static final String[]	CONFIG_DEF_SCHEMA

Constructor Summary

Constructors

Constructor	Description
ColBertEmbedderConfig(ColBertEmbedderConfig.Builder builder)

Method Summary

Modifier and Type	Method	Description
int	documentTokenId()
static String	getDefMd5()
static String	getDefName()
static String	getDefNamespace()
int	maxDocumentTokens()
int	maxQueryTokens()
int	queryTokenId()
Path	tokenizerPath()
com.yahoo.config.ModelReference	tokenizerPathReference()
String	transformerAttentionMask()
int	transformerEndSequenceToken()
ColBertEmbedderConfig.TransformerExecutionMode.Enum	transformerExecutionMode()
int	transformerGpuDevice()
String	transformerInputIds()
int	transformerInterOpThreads()
int	transformerIntraOpThreads()
int	transformerMaskToken()
int	transformerMaxTokens()
Path	transformerModel()
com.yahoo.config.ModelReference	transformerModelReference()
String	transformerOutput()
int	transformerPadToken()
int	transformerStartSequenceToken()

Methods inherited from class com.yahoo.config.ConfigInstance

getDefName, getDefNamespace, serialize, serialize

Methods inherited from class com.yahoo.config.InnerNode

equals, getAllDescendantLeafNodes, getChildren, getChildrenWithVectorsFlattened, hashCode, postInitialize, toString

Methods inherited from class com.yahoo.config.Node

clone

Methods inherited from class java.lang.Object

finalize, getClass, notify, notifyAll, wait, wait, wait

Field Details

CONFIG_DEF_MD5

public static final String CONFIG_DEF_MD5

See Also:

• Constant Field Values

CONFIG_DEF_NAME

public static final String CONFIG_DEF_NAME

See Also:

• Constant Field Values

CONFIG_DEF_NAMESPACE

public static final String CONFIG_DEF_NAMESPACE

See Also:

• Constant Field Values

CONFIG_DEF_SCHEMA

public static final String[] CONFIG_DEF_SCHEMA

Constructor Details

ColBertEmbedderConfig

public ColBertEmbedderConfig(ColBertEmbedderConfig.Builder builder)

Method Details

getDefMd5

public static String getDefMd5()

getDefName

public static String getDefName()

getDefNamespace

public static String getDefNamespace()

tokenizerPath

public Path tokenizerPath()

Returns:

col-bert-embedder.tokenizerPath

tokenizerPathReference

public com.yahoo.config.ModelReference tokenizerPathReference()

Returns:

col-bert-embedder.tokenizerPath ModelReference

transformerModel

public Path transformerModel()

Returns:

col-bert-embedder.transformerModel

transformerModelReference

public com.yahoo.config.ModelReference transformerModelReference()

Returns:

col-bert-embedder.transformerModel ModelReference

maxQueryTokens

public int maxQueryTokens()

Returns:

col-bert-embedder.maxQueryTokens

maxDocumentTokens

public int maxDocumentTokens()

Returns:

col-bert-embedder.maxDocumentTokens

transformerMaxTokens

public int transformerMaxTokens()

Returns:

col-bert-embedder.transformerMaxTokens

transformerInputIds

public String transformerInputIds()

Returns:

col-bert-embedder.transformerInputIds

transformerAttentionMask

public String transformerAttentionMask()

Returns:

col-bert-embedder.transformerAttentionMask

transformerStartSequenceToken

public int transformerStartSequenceToken()

Returns:

col-bert-embedder.transformerStartSequenceToken

transformerEndSequenceToken

public int transformerEndSequenceToken()

Returns:

col-bert-embedder.transformerEndSequenceToken

transformerMaskToken

public int transformerMaskToken()

Returns:

col-bert-embedder.transformerMaskToken

transformerPadToken

public int transformerPadToken()

Returns:

col-bert-embedder.transformerPadToken

queryTokenId

public int queryTokenId()

Returns:

col-bert-embedder.queryTokenId

documentTokenId

public int documentTokenId()

Returns:

col-bert-embedder.documentTokenId

transformerOutput

public String transformerOutput()

Returns:

col-bert-embedder.transformerOutput

transformerExecutionMode

public ColBertEmbedderConfig.TransformerExecutionMode.Enum transformerExecutionMode()

Returns:

col-bert-embedder.transformerExecutionMode

transformerInterOpThreads

public int transformerInterOpThreads()

Returns:

col-bert-embedder.transformerInterOpThreads

transformerIntraOpThreads

public int transformerIntraOpThreads()

Returns:

col-bert-embedder.transformerIntraOpThreads

transformerGpuDevice

public int transformerGpuDevice()

Returns:

col-bert-embedder.transformerGpuDevice