summaryrefslogtreecommitdiffstats
path: root/model-integration
diff options
context:
space:
mode:
authorJon Bratseth <bratseth@gmail.com>2022-09-02 22:13:33 +0200
committerJon Bratseth <bratseth@gmail.com>2022-09-02 22:13:33 +0200
commit9b030fee55a10d0c90acb0d82bb4289f46420262 (patch)
tree1eccedc5fc3cbbe26157a75da3542a5905aef90c /model-integration
parentf519dbc42c68770bfbcff7d11cede53b11ad1751 (diff)
Set as resolved in test
Diffstat (limited to 'model-integration')
-rw-r--r--model-integration/src/test/java/ai/vespa/embedding/BertBaseEmbedderTest.java4
1 files changed, 2 insertions, 2 deletions
diff --git a/model-integration/src/test/java/ai/vespa/embedding/BertBaseEmbedderTest.java b/model-integration/src/test/java/ai/vespa/embedding/BertBaseEmbedderTest.java
index 82a78115e63..dbcae24b28f 100644
--- a/model-integration/src/test/java/ai/vespa/embedding/BertBaseEmbedderTest.java
+++ b/model-integration/src/test/java/ai/vespa/embedding/BertBaseEmbedderTest.java
@@ -23,8 +23,8 @@ public class BertBaseEmbedderTest {
assumeTrue(OnnxEvaluator.isRuntimeAvailable(modelPath));
BertBaseEmbedderConfig.Builder builder = new BertBaseEmbedderConfig.Builder();
- builder.tokenizerVocab(ModelReference.fromPath(vocabPath));
- builder.transformerModel(ModelReference.fromPath(modelPath));
+ builder.tokenizerVocab(ModelReference.valueOf(vocabPath));
+ builder.transformerModel(ModelReference.valueOf(modelPath));
BertBaseEmbedder embedder = new BertBaseEmbedder(builder.build());
TensorType destType = TensorType.fromSpec("tensor<float>(x[7])");