diff options
author | Jo Kristian Bergum <bergum@yahooinc.com> | 2023-12-17 08:48:45 +0100 |
---|---|---|
committer | Jo Kristian Bergum <bergum@yahooinc.com> | 2023-12-17 08:48:45 +0100 |
commit | 745a8db7a8eaea7aa53736a26d64e97543900343 (patch) | |
tree | 3dc56fe5f2b0d0a300cd24470912f98c8842985f /config-model | |
parent | 56ff2f5e971a26d81cfe5cdbac65d856118820e4 (diff) |
Allow mapped 1d tensor for embed expressions
Diffstat (limited to 'config-model')
-rw-r--r-- | config-model/src/main/java/com/yahoo/vespa/model/container/component/SpladeEmbedder.java | 4 |
1 files changed, 0 insertions, 4 deletions
diff --git a/config-model/src/main/java/com/yahoo/vespa/model/container/component/SpladeEmbedder.java b/config-model/src/main/java/com/yahoo/vespa/model/container/component/SpladeEmbedder.java index 96554e91d38..038a6cb78c8 100644 --- a/config-model/src/main/java/com/yahoo/vespa/model/container/component/SpladeEmbedder.java +++ b/config-model/src/main/java/com/yahoo/vespa/model/container/component/SpladeEmbedder.java @@ -13,7 +13,6 @@ import static com.yahoo.vespa.model.container.ContainerModelEvaluation.INTEGRATI public class SpladeEmbedder extends TypedComponent implements SpladeEmbedderConfig.Producer { private final OnnxModelOptions onnxModelOptions; - private final ModelReference modelRef; private final ModelReference vocabRef; private final Integer maxTokens; @@ -21,7 +20,6 @@ public class SpladeEmbedder extends TypedComponent implements SpladeEmbedderConf private final String transformerAttentionMask; private final String transformerTokenTypeIds; private final String transformerOutput; - private final Double termScoreThreshold; public SpladeEmbedder(ApplicationContainerCluster cluster, Element xml, DeployState state) { @@ -52,8 +50,6 @@ public class SpladeEmbedder extends TypedComponent implements SpladeEmbedderConf } throw new IllegalArgumentException("'tokenizer-model' must be specified"); } - - @Override public void getConfig(SpladeEmbedderConfig.Builder b) { b.transformerModel(modelRef).tokenizerPath(vocabRef); |