recipes/AceMath-7B-Instruct/beam_search.yaml (9 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
filter_duplicates: true
approach: beam_search
n: 4
search_batch_size: 1 # DO NOT CHANGE!
num_samples: 10 # REMOVE THIS LINE TO RUN ON THE WHOLE DATASET
seed: 0