diff options
author | Jon Bratseth <bratseth@vespa.ai> | 2023-04-18 11:12:53 +0200 |
---|---|---|
committer | Jon Bratseth <bratseth@vespa.ai> | 2023-04-18 11:12:53 +0200 |
commit | 836058aedf1dc51bd2d5d12199a5934ed410fcac (patch) | |
tree | 4b5d88e69f4555208a02fb3fda779a02a96600a9 /model-integration | |
parent | 5082e42350a117c46f045a31c3502f2c8ad97a0d (diff) |
Pull endtoken
Diffstat (limited to 'model-integration')
-rw-r--r-- | model-integration/src/main/java/ai/vespa/embedding/BertBaseEmbedder.java | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/model-integration/src/main/java/ai/vespa/embedding/BertBaseEmbedder.java b/model-integration/src/main/java/ai/vespa/embedding/BertBaseEmbedder.java index 19536f3cb32..bf56d233f89 100644 --- a/model-integration/src/main/java/ai/vespa/embedding/BertBaseEmbedder.java +++ b/model-integration/src/main/java/ai/vespa/embedding/BertBaseEmbedder.java @@ -48,7 +48,7 @@ public class BertBaseEmbedder extends AbstractComponent implements Embedder { public BertBaseEmbedder(OnnxRuntime onnx, BertBaseEmbedderConfig config) { maxTokens = config.transformerMaxTokens(); startSequenceToken = config.transformerStartSequenceToken(); - endSequenceToken = config.transformerStartSequenceToken(); + endSequenceToken = config.transformerEndSequenceToken(); inputIdsName = config.transformerInputIds(); attentionMaskName = config.transformerAttentionMask(); tokenTypeIdsName = config.transformerTokenTypeIds(); |