diff options
author | Jon Bratseth <bratseth@gmail.com> | 2022-09-02 22:13:33 +0200 |
---|---|---|
committer | Jon Bratseth <bratseth@gmail.com> | 2022-09-02 22:13:33 +0200 |
commit | 9b030fee55a10d0c90acb0d82bb4289f46420262 (patch) | |
tree | 1eccedc5fc3cbbe26157a75da3542a5905aef90c /model-integration | |
parent | f519dbc42c68770bfbcff7d11cede53b11ad1751 (diff) |
Set as resolved in test
Diffstat (limited to 'model-integration')
-rw-r--r-- | model-integration/src/test/java/ai/vespa/embedding/BertBaseEmbedderTest.java | 4 |
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])"); |