From c133e41a23186fe8420417c84fd23f6a33996163 Mon Sep 17 00:00:00 2001 From: Jintao Date: Thu, 9 Jan 2025 23:51:34 +0800 Subject: [PATCH] fix infer engine (#2898) --- swift/llm/infer/infer_engine/infer_engine.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/swift/llm/infer/infer_engine/infer_engine.py b/swift/llm/infer/infer_engine/infer_engine.py index d1c50c0bb..33ca08088 100644 --- a/swift/llm/infer/infer_engine/infer_engine.py +++ b/swift/llm/infer/infer_engine/infer_engine.py @@ -159,7 +159,7 @@ def _gen_wrapper(): result += res i += max_batch_size prog_bar.update(len(tasks_samples)) - return self._update_metrics(res, metrics) + return self._update_metrics(result, metrics) def _get_toolcall(self, response: Union[str, List[Dict[str, Any]]],