mirror of
https://github.com/deepseek-ai/DeepSeek-Prover-V1.5
synced 2025-06-26 18:15:55 +00:00
upload files
This commit is contained in:
31
configs/sampling.py
Normal file
31
configs/sampling.py
Normal file
@@ -0,0 +1,31 @@
|
||||
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,
|
||||
)
|
||||
Reference in New Issue
Block a user