summaryrefslogtreecommitdiffstats
path: root/config-model
diff options
context:
space:
mode:
authorHarald Musum <musum@verizonmedia.com>2023-09-25 21:35:46 +0200
committerGitHub <noreply@github.com>2023-09-25 21:35:46 +0200
commitcdb6e9f363e5b88577d4df3fb98a74ae5399eb14 (patch)
treecb02ab0e1d36efea68ca26aeb3796829ffa6be83 /config-model
parentb4af421142168c36cc1e8c9bae735731a68fcb20 (diff)
parent8f010aec706f1c53b5d2d6775ddcbd61e234ae2e (diff)
Merge pull request #28656 from vespa-engine/hmusum/fix-onnx-model-cost
Try to add back method removed from API but still in use
Diffstat (limited to 'config-model')
-rw-r--r--config-model/src/main/java/com/yahoo/vespa/model/DefaultOnnxModelCost.java2
1 files changed, 1 insertions, 1 deletions
diff --git a/config-model/src/main/java/com/yahoo/vespa/model/DefaultOnnxModelCost.java b/config-model/src/main/java/com/yahoo/vespa/model/DefaultOnnxModelCost.java
index 9794cfe4ad7..fddf8409376 100644
--- a/config-model/src/main/java/com/yahoo/vespa/model/DefaultOnnxModelCost.java
+++ b/config-model/src/main/java/com/yahoo/vespa/model/DefaultOnnxModelCost.java
@@ -56,7 +56,7 @@ public class DefaultOnnxModelCost implements OnnxModelCost {
String path = f.getPath().getRelative();
if (alreadyAnalyzed(path)) return;
log.log(Level.FINE, () -> "Register model '%s'".formatted(path));
- if (f.exists()) {
+ if (f.exists() && appPkg != null) {
var memoryStats = OnnxModelProbe.probeMemoryStats(appPkg, f.getPath()).orElse(null);
if (memoryStats != null) {
log.log(Level.FINE, () -> "Register model '%s' with memory stats: %s".formatted(path, memoryStats));