diff --git a/wlm/src/main/java/ai/djl/serving/wlm/ModelInfo.java b/wlm/src/main/java/ai/djl/serving/wlm/ModelInfo.java
index 80d3c0596bd..70b3e8b7f42 100644
--- a/wlm/src/main/java/ai/djl/serving/wlm/ModelInfo.java
+++ b/wlm/src/main/java/ai/djl/serving/wlm/ModelInfo.java
@@ -779,7 +779,8 @@ public String[] getLoadOnDevices() {
                         gpuCount = procs;
                     }
                 } else if ("DeepSpeed".equals(engineName)
-                        || "FasterTransformer".equals(engineName)) {
+                        || "FasterTransformer".equals(engineName)
+                        || "MPI".equals(engineName)) {
                     return new String[] {"0"};
                 }