[api] Set arguments in serving.properties as model properties (#1853)

Change-Id: I777e04d723afd0edf0dc7227ac6b6abf933a9d81
deepjavalibrary · Aug 2, 2022 · f1ebbe8 · f1ebbe8
1 parent eb1197b
commit f1ebbe8
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;
  }