mirror of
https://github.com/deepseek-ai/DeepSeek-Prover-V1.5
synced 2024-11-21 19:07:44 +00:00
32 lines
568 B
Python
32 lines
568 B
Python
|
from prover.utils import AttrDict
|
||
|
|
from prover.algorithms import Sampling
|
||
|
|
|
||
|
|
|
||
|
|
# dataset
|
||
|
|
data_path = 'datasets/minif2f.jsonl'
|
||
|
|
data_split = ['valid', 'test']
|
||
|
|
data_repeat = 1
|
||
|
|
|
||
|
|
# verifier
|
||
|
|
lean_max_concurrent_requests = 64
|
||
|
|
lean_memory_limit = 10
|
||
|
|
lean_timeout = 300
|
||
|
|
|
||
|
|
# model
|
||
|
|
batch_size = 32
|
||
|
|
model_path = 'deepseek-ai/DeepSeek-Prover-V1.5-RL'
|
||
|
|
model_args = AttrDict(
|
||
|
|
mode='cot', # `cot` or `non-cot`
|
||
|
|
temperature=1,
|
||
|
|
max_tokens=2048,
|
||
|
|
top_p=0.95,
|
||
|
|
)
|
||
|
|
|
||
|
|
# algorithm
|
||
|
|
n_search_procs = 64
|
||
|
|
sampler = dict(
|
||
|
|
algorithm=Sampling,
|
||
|
|
sample_num=128,
|
||
|
|
log_interval=32,
|
||
|
|
)
|