update remove file.

shibing624 · Sep 8, 2023 · 75fa4c4 · 75fa4c4
1 parent 8b90c59
commit 75fa4c4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/examples/gpt/inference_multigpu_demo.py b/examples/gpt/inference_multigpu_demo.py
@@ -140,7 +140,7 @@ def main():
         repetition_penalty=args.repetition_penalty,
     )
     stop_str = tokenizer.eos_token if tokenizer.eos_token else prompt_template.stop_str
-    if os.path.exists(args.output_file):
+    if local_rank <= 0 and os.path.exists(args.output_file):
         os.remove(args.output_file)
     count = 0
     for batch in tqdm(