[wlm] Fixes WorkerThread name (deepjavalibrary#817)

KexinFeng · Aug 16, 2023 · 64202c8 · 64202c8
1 parent 6d9b759
commit 64202c8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/wlm/src/main/java/ai/djl/serving/wlm/WorkerThread.java b/wlm/src/main/java/ai/djl/serving/wlm/WorkerThread.java
@@ -67,7 +67,7 @@ private WorkerThread(Builder<I, O> builder) {
         ZooModel<I, O> model = builder.model.getModel(device);
 
         predictor = model.newPredictor();
-        modelName = model.getName();
+        modelName = builder.model.getId();
         logModelMetric = Boolean.parseBoolean(model.getProperty("log_model_metric"));
         metricsAggregation = Integer.parseInt(model.getProperty("metrics_aggregation", "1000"));
     }