diff --git a/serving/src/main/java/ai/djl/serving/workflow/function/WlmWorkflowFunction.java b/serving/src/main/java/ai/djl/serving/workflow/function/WlmWorkflowFunction.java index 42b282677..1521cac6f 100644 --- a/serving/src/main/java/ai/djl/serving/workflow/function/WlmWorkflowFunction.java +++ b/serving/src/main/java/ai/djl/serving/workflow/function/WlmWorkflowFunction.java @@ -53,7 +53,7 @@ public CompletableFuture run( } return evaluateArgs(args) - .thenComposeAsync( + .thenCompose( processedArgs -> executor.getWlm() .runJob(