recipes/AceMath-7B-Instruct/best_of_n.yaml (10 lines of code) (raw):

# refer to src/sal/config.py for more options model_path: nvidia/AceMath-7B-Instruct prm_path: Qwen/Qwen2.5-Math-PRM-7B custom_chat_template: null approach: best_of_n n: 4 search_batch_size: 25 sort_completed: true filter_duplicates: true num_samples: 10 # REMOVE THIS LINE TO RUN ON THE WHOLE DATASET seed: 0