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

frankfliu · web-flow · commit 9902e9894a48 · 2024-05-30T15:12:11.000-07:00
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;
         }

Original file line number	Diff line number	Diff line change
`@@ -137,7 +137,8 @@ static void convertTrtLLM(ModelInfo<?, ?> info) throws IOException {`
`137`	`137`
`138`	`138`	`static void convertOnnxModel(ModelInfo<?, ?> info) throws IOException {`
`139`	`139`	`String prefix = info.prop.getProperty("option.modelName", info.modelDir.toFile().getName());`
`140`		`- if (Files.isRegularFile(info.modelDir.resolve(prefix + ".onnx"))`
	`140`	`+ if (Files.isRegularFile(info.modelDir)`
	`141`	`+ \|\| Files.isRegularFile(info.modelDir.resolve(prefix + ".onnx"))`
`141`	`142`	`\|\| Files.isRegularFile(info.modelDir.resolve("model.onnx"))) {`
`142`	`143`	`return;`
`143`	`144`	`}`