Skip to content

Commit 627cd39

Browse files
authored
[Evaluation] Bug: Multiple GPUs attempt to save in the same folder (#1567)
1 parent d20994a commit 627cd39

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

src/oumi/core/evaluation/evaluator.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -27,6 +27,7 @@
2727
LMHarnessTaskParams,
2828
)
2929
from oumi.core.configs.params.evaluation_params import EvaluationBackend
30+
from oumi.core.distributed import is_world_process_zero
3031
from oumi.core.evaluation.backends.alpaca_eval import evaluate as evaluate_alpaca_eval
3132
from oumi.core.evaluation.backends.lm_harness import evaluate as evaluate_lm_harness
3233
from oumi.core.evaluation.evaluation_result import EvaluationResult
@@ -186,7 +187,7 @@ def evaluate_task(
186187
evaluation_result.start_time = start_time_str
187188

188189
# Save the output, if an output directory has been provided.
189-
if config.output_dir:
190+
if config.output_dir and is_world_process_zero():
190191
self.save_output(
191192
task_params=task_params,
192193
evaluation_result=evaluation_result,

0 commit comments

Comments
 (0)