[api] Fix saveModel NPE bug (#989)

Change-Id: I288d6ca80a7e987f2e527e18f020fee0f75d8156
deepjavalibrary · May 30, 2021 · 6dc2949 · 6dc2949
1 parent 0e4c162
commit 6dc2949
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/api/src/main/java/ai/djl/BaseModel.java b/api/src/main/java/ai/djl/BaseModel.java
@@ -228,6 +228,9 @@ protected Block loadFromBlockFactory() {
     /** {@inheritDoc} */
     @Override
     public void save(Path modelPath, String newModelName) throws IOException {
+        if (newModelName == null || newModelName.isEmpty()) {
+            newModelName = modelName;
+        }
         if (Files.notExists(modelPath)) {
             Files.createDirectories(modelPath);
         }