File tree Expand file tree Collapse file tree 1 file changed +2
-1
lines changed Expand file tree Collapse file tree 1 file changed +2
-1
lines changed Original file line number Diff line number Diff line change 27
27
LMHarnessTaskParams ,
28
28
)
29
29
from oumi .core .configs .params .evaluation_params import EvaluationBackend
30
+ from oumi .core .distributed import is_world_process_zero
30
31
from oumi .core .evaluation .backends .alpaca_eval import evaluate as evaluate_alpaca_eval
31
32
from oumi .core .evaluation .backends .lm_harness import evaluate as evaluate_lm_harness
32
33
from oumi .core .evaluation .evaluation_result import EvaluationResult
@@ -186,7 +187,7 @@ def evaluate_task(
186
187
evaluation_result .start_time = start_time_str
187
188
188
189
# 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 () :
190
191
self .save_output (
191
192
task_params = task_params ,
192
193
evaluation_result = evaluation_result ,
You can’t perform that action at this time.
0 commit comments