aboutsummaryrefslogtreecommitdiffstats
path: root/config-model/src
diff options
context:
space:
mode:
authorLester Solbakken <lesters@oath.com>2023-08-29 11:21:14 +0200
committerLester Solbakken <lesters@oath.com>2023-08-29 11:21:14 +0200
commit750035a8b4ebcf63dc761a4ffcdcbee610636e4f (patch)
treea5f46d1afa1965e6e4c94c10350aeb478fa38b49 /config-model/src
parent3613451e6e00081cb1294ed35ab5c79e62a5d304 (diff)
Add warning when trying to configure missing ONNX model
Diffstat (limited to 'config-model/src')
-rw-r--r--config-model/src/main/java/com/yahoo/vespa/model/container/xml/ContainerModelBuilder.java9
1 files changed, 7 insertions, 2 deletions
diff --git a/config-model/src/main/java/com/yahoo/vespa/model/container/xml/ContainerModelBuilder.java b/config-model/src/main/java/com/yahoo/vespa/model/container/xml/ContainerModelBuilder.java
index 80b676159cb..e7692aeee7b 100644
--- a/config-model/src/main/java/com/yahoo/vespa/model/container/xml/ContainerModelBuilder.java
+++ b/config-model/src/main/java/com/yahoo/vespa/model/container/xml/ContainerModelBuilder.java
@@ -755,8 +755,13 @@ public class ContainerModelBuilder extends ConfigModelBuilder<ContainerModel> {
Element modelsElement = XML.getChild(onnxElement, "models");
for (Element modelElement : XML.getChildren(modelsElement, "model") ) {
OnnxModel onnxModel = profiles.getOnnxModels().asMap().get(modelElement.getAttribute("name"));
- if (onnxModel == null)
- continue; // Skip if model is not found
+ if (onnxModel == null) {
+ String availableModels = String.join(", ", profiles.getOnnxModels().asMap().keySet());
+ context.getDeployState().getDeployLogger().logApplicationPackage(WARNING,
+ "Model '" + modelElement.getAttribute("name") + "' not found. Available ONNX " +
+ "models are: " + availableModels + ". Skipping this configuration.");
+ continue;
+ }
onnxModel.setStatelessExecutionMode(getStringValue(modelElement, "execution-mode", null));
onnxModel.setStatelessInterOpThreads(getIntValue(modelElement, "interop-threads", -1));
onnxModel.setStatelessIntraOpThreads(getIntValue(modelElement, "intraop-threads", -1));