diff options
author | Harald Musum <musum@yahooinc.com> | 2023-11-20 15:17:28 +0100 |
---|---|---|
committer | Harald Musum <musum@yahooinc.com> | 2023-11-20 15:17:28 +0100 |
commit | 9d28a47b003f5498bc59bfd10017dd55fc7ab6e0 (patch) | |
tree | add8988cb32377e8395c36d2fb46cc9259b5dd63 /config-model/src/main/java/com/yahoo/schema | |
parent | e8c0a04b67b632ea3f98327d8f39cd0293ad8581 (diff) |
Register model with onnx model options
Diffstat (limited to 'config-model/src/main/java/com/yahoo/schema')
-rw-r--r-- | config-model/src/main/java/com/yahoo/schema/OnnxModel.java | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/config-model/src/main/java/com/yahoo/schema/OnnxModel.java b/config-model/src/main/java/com/yahoo/schema/OnnxModel.java index 867ffdb3960..9456baafd57 100644 --- a/config-model/src/main/java/com/yahoo/schema/OnnxModel.java +++ b/config-model/src/main/java/com/yahoo/schema/OnnxModel.java @@ -1,9 +1,9 @@ // Copyright Vespa.ai. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root. package com.yahoo.schema; +import com.yahoo.config.model.api.OnnxModelOptions; import com.yahoo.searchlib.rankingexpression.Reference; import com.yahoo.tensor.TensorType; -import com.yahoo.vespa.model.container.component.OnnxModelOptions; import com.yahoo.vespa.model.ml.OnnxModelInfo; import java.util.Collections; @@ -171,4 +171,6 @@ public class OnnxModel extends DistributableResource implements Cloneable { return onnxModelOptions.gpuDevice(); } + public OnnxModelOptions onnxModelOptions() { return onnxModelOptions; } + } |