diff options
author | Jon Bratseth <bratseth@gmail.com> | 2022-09-15 10:41:48 +0200 |
---|---|---|
committer | GitHub <noreply@github.com> | 2022-09-15 10:41:48 +0200 |
commit | 641561888e998be6d198c651c3c45de9f1915b31 (patch) | |
tree | 1b94433c10755a3d02cac6c8183754614a909869 /model-integration/src/main/java/ai | |
parent | e430aa42191d81b0f0aca9a0c1403575eaea56b1 (diff) | |
parent | 8d588c26e716d705c4f1b63fc6b16ce17eed0647 (diff) |
Merge pull request #24058 from vespa-engine/bratseth/mode-def-file
Move the BERT config def to model-integration
Diffstat (limited to 'model-integration/src/main/java/ai')
-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 c25fc8704b0..d4a93999dff 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())); } } |