[Misc] improve example script output (#15528)

Signed-off-by: reidliu41 <reid201711@gmail.com> Co-authored-by: reidliu41 <reid201711@gmail.com>
2025-03-26 18:12:47 +08:00
parent 99f536f830
commit 4ec2cee000
5 changed files with 16 additions and 6 deletions
--- a/examples/offline_inference/basic/embed.py
+++ b/examples/offline_inference/basic/embed.py
@@ -23,12 +23,14 @@ def main(args: Namespace):
    outputs = model.embed(prompts)

    # Print the outputs.
+    print("\nGenerated Outputs:\n" + "-" * 60)
    for prompt, output in zip(prompts, outputs):
        embeds = output.outputs.embedding
        embeds_trimmed = ((str(embeds[:16])[:-1] +
                           ", ...]") if len(embeds) > 16 else embeds)
-        print(f"Prompt: {prompt!r} | "
+        print(f"Prompt: {prompt!r} \n"
              f"Embeddings: {embeds_trimmed} (size={len(embeds)})")
+        print("-" * 60)


 if __name__ == "__main__":