[bugfix] parsing waiting steps to integer (#1178)

deepjavalibrary · Oct 12, 2023 · 9f0d840 · 9f0d840
1 parent 6065232
commit 9f0d840
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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: