diff options
author | Martin Polden <mpolden@mpolden.no> | 2023-01-26 14:42:15 +0100 |
---|---|---|
committer | Martin Polden <mpolden@mpolden.no> | 2023-01-26 14:42:15 +0100 |
commit | 3d4f199ed58cce65680c775df824ae376e518a23 (patch) | |
tree | ba5cb7bc03ccbdbc0cdc724f63c031c2683df309 /model-evaluation | |
parent | ed5d394998b5538e2a3330409391ab18acadb1df (diff) |
Remove 'required' attribute
Diffstat (limited to 'model-evaluation')
-rw-r--r-- | model-evaluation/src/main/java/ai/vespa/models/evaluation/RankProfilesConfigImporter.java | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/model-evaluation/src/main/java/ai/vespa/models/evaluation/RankProfilesConfigImporter.java b/model-evaluation/src/main/java/ai/vespa/models/evaluation/RankProfilesConfigImporter.java index 9877dd69e83..924eed18633 100644 --- a/model-evaluation/src/main/java/ai/vespa/models/evaluation/RankProfilesConfigImporter.java +++ b/model-evaluation/src/main/java/ai/vespa/models/evaluation/RankProfilesConfigImporter.java @@ -182,7 +182,7 @@ public class RankProfilesConfigImporter { options.setExecutionMode(onnxModelConfig.stateless_execution_mode()); options.setInterOpThreads(onnxModelConfig.stateless_interop_threads()); options.setIntraOpThreads(onnxModelConfig.stateless_intraop_threads()); - options.setGpuDevice(onnxModelConfig.gpu_device(), onnxModelConfig.gpu_device_required()); + options.setGpuDevice(onnxModelConfig.gpu_device()); return new OnnxModel(name, file, options); } catch (InterruptedException e) { throw new IllegalStateException("Gave up waiting for ONNX model " + onnxModelConfig.name()); |