summaryrefslogtreecommitdiffstats
path: root/config-model/src
diff options
context:
space:
mode:
authorLester Solbakken <lesters@users.noreply.github.com>2023-08-29 12:30:36 +0200
committerGitHub <noreply@github.com>2023-08-29 12:30:36 +0200
commit639deae84372d6f38944cf9a6b663ce8924a3bd7 (patch)
treea3bbd24e173b3b3ecd74ad45542315709129cf8a /config-model/src
parent2aec4066094c19660913414cbb53c976185cc4a7 (diff)
parent750035a8b4ebcf63dc761a4ffcdcbee610636e4f (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.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));