fix infer engine (#2898)

modelscope · Jan 9, 2025 · c133e41 · c133e41
1 parent 005674f
commit c133e41
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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]]],