[api] Set arguments in serving.properties as model properties

Change-Id: I777e04d723afd0edf0dc7227ac6b6abf933a9d81
deepjavalibrary · Aug 2, 2022 · f92ef2c · f92ef2c
1 parent eb1197b
commit f92ef2c
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/api/src/main/java/ai/djl/repository/zoo/BaseModelLoader.java b/api/src/main/java/ai/djl/repository/zoo/BaseModelLoader.java
@@ -200,6 +200,9 @@ protected Model createModel(
         if (block != null) {
             model.setBlock(block);
         }
+        for (Map.Entry<String, Object> entry : arguments.entrySet()) {
+            model.setProperty(entry.getKey(), entry.getValue().toString());
+        }
         return model;
     }