text-generation: improve output printing (huggingface#1486)

HabanaAI · Jan 20, 2025 · 27a5da5 · 27a5da5
1 parent be15e40
commit 27a5da5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/examples/text-generation/run_generation.py b/examples/text-generation/run_generation.py
@@ -569,7 +569,7 @@ def rounder(x):
             for j, output in enumerate(
                 zip(generated[args.num_return_sequences * i : args.num_return_sequences * (i + 1)])
             ):
-                print(f"output {j+1}: {output}")
+                print(f"output {i+1}.{j+1}: {output}")
                 all_outputs.append(output)
             print()