aboutsummaryrefslogtreecommitdiffstats
path: root/model-integration/src/main/java
diff options
context:
space:
mode:
authorJon Bratseth <bratseth@gmail.com>2022-09-14 17:51:12 +0200
committerJon Bratseth <bratseth@gmail.com>2022-09-14 17:51:12 +0200
commitc172d09ebb17b47ebb2e75bb04d09a8533350450 (patch)
tree52b8f915bd2bc1352c05912b7f9597eb2b7ba73b /model-integration/src/main/java
parent027390f4a619bee2a713610f5057e79f2c1eb36c (diff)
Move the BERT config def to model-integration
Diffstat (limited to 'model-integration/src/main/java')
-rw-r--r--model-integration/src/main/java/ai/vespa/embedding/BertBaseEmbedder.java4
1 files changed, 2 insertions, 2 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..4a1f59cc5cf 100644
--- a/model-integration/src/main/java/ai/vespa/embedding/BertBaseEmbedder.java
+++ b/model-integration/src/main/java/ai/vespa/embedding/BertBaseEmbedder.java
@@ -17,7 +17,7 @@ import java.util.Map;
/**
* A BERT Base compatible embedder. This embedder uses a WordPiece embedder to
- * produce a token sequence that is input to a transformer model. A BERT base
+ * produce a token sequence that is then input to a transformer model. A BERT base
* compatible transformer model must have three inputs:
*
* - A token sequence (input_ids)
@@ -76,7 +76,7 @@ public class BertBaseEmbedder implements Embedder {
private void validateName(Map<String, TensorType> types, String name, String type) {
if ( ! types.containsKey(name)) {
throw new IllegalArgumentException("Model does not contain required " + type + ": '" + name + "'. " +
- "Model contains: " + String.join(",", types.keySet()));
+ "Model contains: " + String.join(",", types.keySet()));
}
}