From c670985a40bdcd967532347a5ff99b8efd1fe1ce Mon Sep 17 00:00:00 2001
From: "xingjun.wang" <xingjun.wxj@alibaba-inc.com>
Date: Mon, 23 Oct 2023 14:37:01 +0800
Subject: [PATCH] update

---
 modelscope/pipelines/nlp/llm_pipeline.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/modelscope/pipelines/nlp/llm_pipeline.py b/modelscope/pipelines/nlp/llm_pipeline.py
index 6138a7a9..90273635 100644
--- a/modelscope/pipelines/nlp/llm_pipeline.py
+++ b/modelscope/pipelines/nlp/llm_pipeline.py
@@ -131,7 +131,7 @@ class LLMPipeline(Pipeline):
         if output_logits:
             logits_d: dict = {
                 'logits': outputs,
-                'tokens': tokens,
+                'tokens': tokens['inputs'],
                 'inputs_len': len(tokens['inputs'][0])
             }
             response.update(logits_d)