fix offline inference chat response prompt (#32088)

Signed-off-by: Andy Xie <andy.xning@gmail.com>
This commit is contained in:
Ning Xie
2026-01-11 22:01:18 +08:00
committed by GitHub
parent a34abc49b7
commit d74132ca3b
2 changed files with 14 additions and 9 deletions

View File

@@ -152,9 +152,12 @@ def main(args):
# print the generated text
if args.print_output:
for output in outputs:
for i, output in enumerate(outputs):
print("-" * 50)
print(f"prompt: {output.prompt}")
if not args.custom_mm_prompts:
print(f"prompt: {prompts[i].prompt}")
else:
print(f"prompt: {prompts[i]}")
print(f"generated text: {output.outputs[0].text}")
print("-" * 50)