diff options
Diffstat (limited to 'model-integration/src/main/java/ai/vespa/modelintegration/evaluator/OnnxEvaluator.java')
-rw-r--r-- | model-integration/src/main/java/ai/vespa/modelintegration/evaluator/OnnxEvaluator.java | 6 |
1 files changed, 5 insertions, 1 deletions
diff --git a/model-integration/src/main/java/ai/vespa/modelintegration/evaluator/OnnxEvaluator.java b/model-integration/src/main/java/ai/vespa/modelintegration/evaluator/OnnxEvaluator.java index c9ab9924214..bdcceddb04f 100644 --- a/model-integration/src/main/java/ai/vespa/modelintegration/evaluator/OnnxEvaluator.java +++ b/model-integration/src/main/java/ai/vespa/modelintegration/evaluator/OnnxEvaluator.java @@ -94,8 +94,12 @@ public class OnnxEvaluator { } public static boolean isRuntimeAvailable() { + return isRuntimeAvailable(""); + } + + public static boolean isRuntimeAvailable(String modelPath) { try { - new OnnxEvaluator(""); + new OnnxEvaluator(modelPath); return true; } catch (UnsatisfiedLinkError | RuntimeException | NoClassDefFoundError e) { return false; |