update

2026-02-24 20:19:51 +01:00 · 2023-10-23 14:37:01 +08:00
parent e340310600
commit c670985a40
1 changed files with 1 additions and 1 deletions
--- 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)