diff options
author | Harald Musum <musum@verizonmedia.com> | 2023-09-25 21:35:46 +0200 |
---|---|---|
committer | GitHub <noreply@github.com> | 2023-09-25 21:35:46 +0200 |
commit | cdb6e9f363e5b88577d4df3fb98a74ae5399eb14 (patch) | |
tree | cb02ab0e1d36efea68ca26aeb3796829ffa6be83 /config-model | |
parent | b4af421142168c36cc1e8c9bae735731a68fcb20 (diff) | |
parent | 8f010aec706f1c53b5d2d6775ddcbd61e234ae2e (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.java | 2 |
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)); |