update

modelscope · Nov 6, 2024 · 8d3d4b0 · 8d3d4b0
1 parent 1ce44c5
commit 8d3d4b0
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/swift/llm/deploy.py b/swift/llm/deploy.py
@@ -516,9 +516,7 @@ async def _generate_full():
             prompt_tokens=num_prompt_tokens,
             completion_tokens=num_generated_tokens,
             total_tokens=num_prompt_tokens + num_generated_tokens)
-        finish_reason = None
-        if output.status.name == 'FINISH':
-            finish_reason = 'stop'
+        finish_reason = 'stop' if output.status.name == 'FINISH' else None
 
         if isinstance(request, ChatCompletionRequest):
             action, action_input = split_action_action_input(response)