From 372bd2c677bb9707c55a9153f860fb2017ce6ffc Mon Sep 17 00:00:00 2001 From: Bjørn Christian Seime Date: Tue, 9 May 2023 09:49:40 +0200 Subject: Reapply "Bjorncs/embedder onnx gpu" Updated model-integration to depend on configdefinitions instead of searchcore as OnnxModelsConfig is moved there. --- .../src/main/java/ai/vespa/models/evaluation/ModelsEvaluator.java | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'model-evaluation') 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, -- cgit v1.2.3