summaryrefslogtreecommitdiffstats
path: root/model-evaluation
diff options
context:
space:
mode:
authorBjørn Christian Seime <bjorncs@yahooinc.com>2023-05-09 09:49:40 +0200
committerBjørn Christian Seime <bjorncs@yahooinc.com>2023-05-09 13:02:43 +0200
commit372bd2c677bb9707c55a9153f860fb2017ce6ffc (patch)
tree064bccc8e96f496e76ac5be148ab2f91e65c5ed7 /model-evaluation
parent9a41de7b23dce838df8a8ebac42fc41da3478bb9 (diff)
Reapply "Bjorncs/embedder onnx gpu"
Updated model-integration to depend on configdefinitions instead of searchcore as OnnxModelsConfig is moved there.
Diffstat (limited to 'model-evaluation')
-rw-r--r--model-evaluation/src/main/java/ai/vespa/models/evaluation/ModelsEvaluator.java2
1 files changed, 1 insertions, 1 deletions
diff --git a/model-evaluation/src/main/java/ai/vespa/models/evaluation/ModelsEvaluator.java b/model-evaluation/src/main/java/ai/vespa/models/evaluation/ModelsEvaluator.java
index fd5306f9add..303d2acd79e 100644
--- a/model-evaluation/src/main/java/ai/vespa/models/evaluation/ModelsEvaluator.java
+++ b/model-evaluation/src/main/java/ai/vespa/models/evaluation/ModelsEvaluator.java
@@ -41,7 +41,7 @@ public class ModelsEvaluator extends AbstractComponent {
RankingExpressionsConfig expressionsConfig,
OnnxModelsConfig onnxModelsConfig,
FileAcquirer fileAcquirer) {
- this(config, constantsConfig, expressionsConfig, onnxModelsConfig, fileAcquirer, new OnnxRuntime());
+ this(config, constantsConfig, expressionsConfig, onnxModelsConfig, fileAcquirer, new OnnxRuntime(onnxModelsConfig));
}
public ModelsEvaluator(RankProfilesConfigImporter importer,