fix the parsing issue (deepjavalibrary#1857)

patins1 · Aug 26, 2022 · ddd126b · ddd126b
1 parent a7ff335
commit ddd126b
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/...es/paddlepaddle/paddlepaddle-engine/src/main/java/ai/djl/paddlepaddle/engine/PpModel.java b/...es/paddlepaddle/paddlepaddle-engine/src/main/java/ai/djl/paddlepaddle/engine/PpModel.java
@@ -19,6 +19,7 @@
 import ai.djl.ndarray.NDManager;
 import ai.djl.ndarray.types.DataType;
 import ai.djl.paddlepaddle.jni.JniUtils;
+import ai.djl.translate.ArgumentsUtil;
 import ai.djl.translate.Translator;
 
 import java.io.FileNotFoundException;
@@ -86,10 +87,12 @@ public void load(Path modelPath, String prefix, Map<String, ?> options) throws I
                 JniUtils.disableGLog(config);
             }
             if (options.containsKey("CMLNumThreads")) {
-                JniUtils.cpuMathLibraryNumThreads(config, (Integer) options.get("CMLNumThreads"));
+                JniUtils.cpuMathLibraryNumThreads(
+                        config, ArgumentsUtil.intValue(options, "CMLNumThreads"));
             }
             if (options.containsKey("SwitchIrOptim")) {
-                JniUtils.switchIrOptim(config, (Boolean) options.get("SwitchIrOptim"));
+                JniUtils.switchIrOptim(
+                        config, ArgumentsUtil.booleanValue(options, "SwitchIrOptim"));
             }
         }
         paddlePredictor = new PaddlePredictor(JniUtils.createPredictor(config));