[wlm] Trim whitespce for model_id (deepjavalibrary#2175)

siddvenk · Jul 15, 2024 · 4a77c76 · 4a77c76
1 parent 1a6eb75
commit 4a77c76
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/wlm/src/main/java/ai/djl/serving/wlm/ModelInfo.java b/wlm/src/main/java/ai/djl/serving/wlm/ModelInfo.java
@@ -703,7 +703,7 @@ private String inferEngine() throws ModelException {
 
         String modelId = prop.getProperty("option.model_id");
         if (modelId != null && modelId.startsWith("djl://")) {
-            Repository repo = Repository.newInstance("tmp", modelId);
+            Repository repo = Repository.newInstance("tmp", modelId.trim());
             MRL mrl = repo.getResources().get(0);
             String groupId = mrl.getGroupId();
             ModelZoo zoo = ModelZoo.getModelZoo(groupId);
@@ -1112,10 +1112,10 @@ void downloadS3() throws ModelException, IOException {
             return;
         }
         if (modelId.startsWith("s3://")) {
-            this.downloadDir = downloadS3ToDownloadDir(modelId);
+            this.downloadDir = downloadS3ToDownloadDir(modelId.trim());
         } else if (modelId.startsWith("djl://")) {
             logger.info("{}: djl model zoo url found: {}", uid, modelId);
-            modelUrl = modelId;
+            modelUrl = modelId.trim();
             // download real model from model zoo
             downloadModel();
         }