args:
{'add_timestamp': True,
'hf_entity': 'vwxyzjn',
'hf_repo_id': 'rejection_sampling',
'hf_repo_id_scores': 'rejection_sampling_scores',
'include_reference_completion_for_rejection_sampling': True,
'input_filename': 'output/completions.jsonl',
'max_forward_batch_size': 64,
'mode': 'judgement'… See the full description on the dataset page:
https://huggingface.co/datasets/vwxyzjn/rejection_sampling_scores_1727887719.