diff options
author | Lester Solbakken <lesters@users.noreply.github.com> | 2023-08-29 12:30:36 +0200 |
---|---|---|
committer | GitHub <noreply@github.com> | 2023-08-29 12:30:36 +0200 |
commit | 639deae84372d6f38944cf9a6b663ce8924a3bd7 (patch) | |
tree | a3bbd24e173b3b3ecd74ad45542315709129cf8a /config-model/src | |
parent | 2aec4066094c19660913414cbb53c976185cc4a7 (diff) | |
parent | 750035a8b4ebcf63dc761a4ffcdcbee610636e4f (diff) |
Merge pull request #28236 from vespa-engine/lesters/warning-missing-onnx-model
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)); |