From 3d4f199ed58cce65680c775df824ae376e518a23 Mon Sep 17 00:00:00 2001 From: Martin Polden Date: Thu, 26 Jan 2023 14:42:15 +0100 Subject: Remove 'required' attribute --- .../java/ai/vespa/models/evaluation/RankProfilesConfigImporter.java | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'model-evaluation/src/main') 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()); -- cgit v1.2.3