diff options
author | Lester Solbakken <lesters@oath.com> | 2023-08-29 11:21:14 +0200 |
---|---|---|
committer | Lester Solbakken <lesters@oath.com> | 2023-08-29 11:21:14 +0200 |
commit | 750035a8b4ebcf63dc761a4ffcdcbee610636e4f (patch) | |
tree | a5f46d1afa1965e6e4c94c10350aeb478fa38b49 /config-model/src | |
parent | 3613451e6e00081cb1294ed35ab5c79e62a5d304 (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.java | 9 |
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)); |