We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 37066d9 commit 9a91aefCopy full SHA for 9a91aef
cookbook/client/tinker/self_host/short_math_grpo.py
@@ -91,7 +91,7 @@ def __call__(self, trajectories: List[Dict[str, Any]], **kwargs) -> List[float]:
91
def create_gsm8k_dataset():
92
"""Create GSM8K dataset."""
93
dataset = Dataset(DatasetMeta('ms://modelscope/gsm8k', subset_name='main', split='train', data_slice=range(DATA_NUM)))
94
- dataset.set_template('Qwen3_5Template', model_id=f'ms://{BASE_MODEL}', max_length=4096,
+ dataset.set_template('Qwen3_5Template', model_id=f'ms://{BASE_MODEL}', max_length=4096,
95
truncation_strategy='delete', enable_thinking=False)
96
dataset.map(GSM8KProcessor(system=SYSTEM_PROMPT))
97
dataset.encode(add_generation_prompt=True)
0 commit comments