Package com.yahoo.embedding

Class SpladeEmbedderConfig

java.lang.Object

com.yahoo.config.Node

com.yahoo.config.InnerNode

com.yahoo.config.ConfigInstance

com.yahoo.embedding.SpladeEmbedderConfig

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

This class represents the root node of splade-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	SpladeEmbedderConfig.Builder
static interface	SpladeEmbedderConfig.Producer
static final class	SpladeEmbedderConfig.TransformerExecutionMode	This class represents splade-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
SpladeEmbedderConfig(SpladeEmbedderConfig.Builder builder)

Method Summary

Modifier and Type	Method	Description
static String	getDefMd5()
static String	getDefName()
static String	getDefNamespace()
double	termScoreThreshold()
Path	tokenizerPath()
com.yahoo.config.ModelReference	tokenizerPathReference()
String	transformerAttentionMask()
SpladeEmbedderConfig.TransformerExecutionMode.Enum	transformerExecutionMode()
int	transformerGpuDevice()
String	transformerInputIds()
int	transformerInterOpThreads()
int	transformerIntraOpThreads()
int	transformerMaxTokens()
Path	transformerModel()
com.yahoo.config.ModelReference	transformerModelReference()
String	transformerOutput()
String	transformerTokenTypeIds()

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

SpladeEmbedderConfig

public SpladeEmbedderConfig(SpladeEmbedderConfig.Builder builder)

Method Details

getDefMd5

public static String getDefMd5()

getDefName

public static String getDefName()

getDefNamespace

public static String getDefNamespace()

tokenizerPath

public Path tokenizerPath()

Returns:

splade-embedder.tokenizerPath

tokenizerPathReference

public com.yahoo.config.ModelReference tokenizerPathReference()

Returns:

splade-embedder.tokenizerPath ModelReference

transformerModel

public Path transformerModel()

Returns:

splade-embedder.transformerModel

transformerModelReference

public com.yahoo.config.ModelReference transformerModelReference()

Returns:

splade-embedder.transformerModel ModelReference

transformerMaxTokens

public int transformerMaxTokens()

Returns:

splade-embedder.transformerMaxTokens

transformerInputIds

public String transformerInputIds()

Returns:

splade-embedder.transformerInputIds

transformerAttentionMask

public String transformerAttentionMask()

Returns:

splade-embedder.transformerAttentionMask

transformerTokenTypeIds

public String transformerTokenTypeIds()

Returns:

splade-embedder.transformerTokenTypeIds

transformerOutput

public String transformerOutput()

Returns:

splade-embedder.transformerOutput

termScoreThreshold

public double termScoreThreshold()

Returns:

splade-embedder.termScoreThreshold

transformerExecutionMode

public SpladeEmbedderConfig.TransformerExecutionMode.Enum transformerExecutionMode()

Returns:

splade-embedder.transformerExecutionMode

transformerInterOpThreads

public int transformerInterOpThreads()

Returns:

splade-embedder.transformerInterOpThreads

transformerIntraOpThreads

public int transformerIntraOpThreads()

Returns:

splade-embedder.transformerIntraOpThreads

transformerGpuDevice

public int transformerGpuDevice()

Returns:

splade-embedder.transformerGpuDevice