diff options
author | Jon Bratseth <bratseth@gmail.com> | 2022-09-14 17:51:12 +0200 |
---|---|---|
committer | Jon Bratseth <bratseth@gmail.com> | 2022-09-14 17:51:12 +0200 |
commit | c172d09ebb17b47ebb2e75bb04d09a8533350450 (patch) | |
tree | 52b8f915bd2bc1352c05912b7f9597eb2b7ba73b /model-integration/src/main/java | |
parent | 027390f4a619bee2a713610f5057e79f2c1eb36c (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.java | 4 |
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())); } } |