From c172d09ebb17b47ebb2e75bb04d09a8533350450 Mon Sep 17 00:00:00 2001 From: Jon Bratseth Date: Wed, 14 Sep 2022 17:51:12 +0200 Subject: Move the BERT config def to model-integration --- .../src/main/java/ai/vespa/embedding/BertBaseEmbedder.java | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) (limited to 'model-integration/src/main/java') 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 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())); } } -- cgit v1.2.3