diff options
author | Jon Bratseth <bratseth@gmail.com> | 2022-09-01 12:59:06 +0200 |
---|---|---|
committer | Jon Bratseth <bratseth@gmail.com> | 2022-09-01 12:59:06 +0200 |
commit | 86a39ab21db4a92b46944709251ccab58ef39370 (patch) | |
tree | 5cac1fc0065019c5161b4000ba28f4382322a664 /model-integration | |
parent | 0652206123f2a65156bf21663aac0e75b4f43b73 (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.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 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(); } |