fix bug with duplicate models when HF_MODEL_ID points to model store (#2054)

siddvenk · web-flow · commit eae8c784b8d6 · 2024-06-12T16:00:29.000-07:00
diff --git a/serving/src/main/java/ai/djl/serving/ModelServer.java b/serving/src/main/java/ai/djl/serving/ModelServer.java
@@ -505,7 +505,7 @@ private String createHuggingFaceModel(String modelId) throws IOException {
         Path path = Paths.get(modelId);
         if (Files.exists(path)) {
             // modelId point to a local file
-            return modelId;
+            return mapModelUrl(path);
         }
 
         // TODO: Download the full model from HF

Original file line number	Diff line number	Diff line change
`@@ -505,7 +505,7 @@ private String createHuggingFaceModel(String modelId) throws IOException {`
`505`	`505`	`Path path = Paths.get(modelId);`
`506`	`506`	`if (Files.exists(path)) {`
`507`	`507`	`// modelId point to a local file`
`508`		`- return modelId;`
	`508`	`+ return mapModelUrl(path);`
`509`	`509`	`}`
`510`	`510`
`511`	`511`	`// TODO: Download the full model from HF`