diff options
-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 149598ee2dd..c25fc8704b0 100644 --- a/model-integration/src/main/java/ai/vespa/embedding/BertBaseEmbedder.java +++ b/model-integration/src/main/java/ai/vespa/embedding/BertBaseEmbedder.java @@ -152,7 +152,7 @@ public class BertBaseEmbedder implements Embedder { } private static Tensor createTokenTypeIds(Tensor d) { - return d.map((x) -> x > 0 ? 0:0); + return d.map((x) -> 0); // Assume only one token type } private int modifyThreadCount(int numThreads) { |