print match rate only on rank 0

microsoft · Jan 15, 2024 · 20f47c7 · 20f47c7
1 parent 246da21
commit 20f47c7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/inference/huggingface/text-generation/ds-hf-compare.py b/inference/huggingface/text-generation/ds-hf-compare.py
@@ -110,4 +110,4 @@ def string_similarity(str1, str2):
         similarity = string_similarity(base_out['generated_text'], ds_out[0]['generated_text'])
         if args.print_outputs: print(f"The similarity ratio is: {similarity*100}%")
 
-print(f"Matches: {match_count}\nMismatches: {mismatch_count}")
+print_0(f"Matches: {match_count}\nMismatches: {mismatch_count}")