args:
{'add_timestamp': False,
'hf_entity': 'jacobmorrison',
'hf_repo_id': 'rejection_sampling_8548',
'hf_repo_id_scores': 'scores_8548',
'input_filename': '/output/shards/8548/2.jsonl',
'max_forward_batch_size': 64,
'mode': 'judgement',
'model_names_or_paths': ['/reward_model'],
'num_completions': 2… See the full description on the dataset page:
https://huggingface.co/datasets/jacobmorrison/rejection_sampling_8548.