diff options
author | Bjørn Christian Seime <bjorncs@yahooinc.com> | 2023-05-08 13:55:00 +0200 |
---|---|---|
committer | Bjørn Christian Seime <bjorncs@yahooinc.com> | 2023-05-08 13:55:00 +0200 |
commit | b4073925d4ce5c08ebc91620219541cb4114ac52 (patch) | |
tree | ee96ae200505f7ca0d6a8cc46855577692fecec8 /model-evaluation | |
parent | 85289c1c179d3469bfe7681ad3d04488185e6c7d (diff) |
Require GPU when available for ONNX evaluation in global-phase and embedders
Diffstat (limited to 'model-evaluation')
-rw-r--r-- | model-evaluation/src/main/java/ai/vespa/models/evaluation/ModelsEvaluator.java | 2 |
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, |