Include only options that control how an Onnx model is loaded in OnnxModelOptions

1 files changed, 2 insertions, 2 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 0d2c695280e..867ffdb3960 100644
--- a/config-model/src/main/java/com/yahoo/schema/OnnxModel.java
+++ b/config-model/src/main/java/com/yahoo/schema/OnnxModel.java
@@ -143,7 +143,7 @@ public class OnnxModel extends DistributableResource implements Cloneable {
 
     public void setStatelessInterOpThreads(int interOpThreads) {
         if (interOpThreads >= 0) {
-            onnxModelOptions = onnxModelOptions.withInteropThreads(interOpThreads);
+            onnxModelOptions = onnxModelOptions.withInterOpThreads(interOpThreads);
         }
     }
 
@@ -153,7 +153,7 @@ public class OnnxModel extends DistributableResource implements Cloneable {
 
     public void setStatelessIntraOpThreads(int intraOpThreads) {
         if (intraOpThreads >= 0) {
-            onnxModelOptions = onnxModelOptions.withIntraopThreads(intraOpThreads);
+            onnxModelOptions = onnxModelOptions.withIntraOpThreads(intraOpThreads);
         }
     }