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

…2054)
deepjavalibrary · Jun 12, 2024 · eae8c78 · eae8c78
1 parent f04d5f3
commit eae8c78
Showing 1 changed file with 1 addition and 1 deletion.
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