summaryrefslogtreecommitdiffstats
path: root/model-integration
diff options
context:
space:
mode:
authorJon Bratseth <bratseth@gmail.com>2022-09-01 12:59:06 +0200
committerJon Bratseth <bratseth@gmail.com>2022-09-01 12:59:06 +0200
commit86a39ab21db4a92b46944709251ccab58ef39370 (patch)
tree5cac1fc0065019c5161b4000ba28f4382322a664 /model-integration
parent0652206123f2a65156bf21663aac0e75b4f43b73 (diff)
Use Path as user type for models
Diffstat (limited to 'model-integration')
-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 3dd8d7eefc4..149598ee2dd 100644
--- a/model-integration/src/main/java/ai/vespa/embedding/BertBaseEmbedder.java
+++ b/model-integration/src/main/java/ai/vespa/embedding/BertBaseEmbedder.java
@@ -57,8 +57,8 @@ public class BertBaseEmbedder implements Embedder {
options.setInterOpThreads(modifyThreadCount(config.onnxInterOpThreads()));
options.setIntraOpThreads(modifyThreadCount(config.onnxIntraOpThreads()));
- tokenizer = new WordPieceEmbedder.Builder(config.tokenizerVocab().value()).build();
- evaluator = new OnnxEvaluator(config.transformerModel().value(), options);
+ tokenizer = new WordPieceEmbedder.Builder(config.tokenizerVocab().toString()).build();
+ evaluator = new OnnxEvaluator(config.transformerModel().toString(), options);
validateModel();
}