[serving] Fixes onnx model conversion bug (#2000)

deepjavalibrary · May 30, 2024 · 9902e98 · 9902e98
1 parent 2bf31b6
commit 9902e98
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/wlm/src/main/java/ai/djl/serving/wlm/LmiUtils.java b/wlm/src/main/java/ai/djl/serving/wlm/LmiUtils.java
@@ -137,7 +137,8 @@ static void convertTrtLLM(ModelInfo<?, ?> info) throws IOException {
 
     static void convertOnnxModel(ModelInfo<?, ?> info) throws IOException {
         String prefix = info.prop.getProperty("option.modelName", info.modelDir.toFile().getName());
-        if (Files.isRegularFile(info.modelDir.resolve(prefix + ".onnx"))
+        if (Files.isRegularFile(info.modelDir)
+                || Files.isRegularFile(info.modelDir.resolve(prefix + ".onnx"))
                 || Files.isRegularFile(info.modelDir.resolve("model.onnx"))) {
             return;
         }