From 9f0d84069ffde8c6e37a25d902b008ebcf0fb481 Mon Sep 17 00:00:00 2001 From: Qing Lan Date: Wed, 11 Oct 2023 21:11:30 -0700 Subject: [PATCH] [bugfix] parsing waiting steps to integer (#1178) --- engines/python/setup/djl_python/huggingface.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/engines/python/setup/djl_python/huggingface.py b/engines/python/setup/djl_python/huggingface.py index c7b0038b1..ec635eb6c 100644 --- a/engines/python/setup/djl_python/huggingface.py +++ b/engines/python/setup/djl_python/huggingface.py @@ -213,7 +213,7 @@ def initialize(self, properties: dict): if "output_formatter" in properties: kwargs["output_formatter"] = properties.get("output_formatter") if "waiting_steps" in properties: - kwargs["waiting_steps"] = properties.get("waiting_steps") + kwargs["waiting_steps"] = int(properties.get("waiting_steps")) self.rolling_batch_type = self.rolling_batch_type.lower() is_mpi = properties.get("engine") != "Python" if is_mpi: