update file exists.

shibing624 · Sep 8, 2023 · 7f06299 · 7f06299
1 parent 6d14add
commit 7f06299
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/inference_multigpu_demo.py b/inference_multigpu_demo.py
@@ -138,7 +138,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(