shank commited on
Commit Β·
cb09ef1
1
Parent(s): 4668456
Fix: Fixed file
Browse files- training/train_grpo.py +1 -1
training/train_grpo.py
CHANGED
|
@@ -477,7 +477,7 @@ trainer = GRPOTrainer(
|
|
| 477 |
args=config,
|
| 478 |
train_dataset=make_dataset(0),
|
| 479 |
reward_funcs=reward_fn,
|
| 480 |
-
|
| 481 |
)
|
| 482 |
|
| 483 |
# ββ Curriculum callback βββββββββββββββββββββββββββββββββββββββββββββββββββββββ
|
|
|
|
| 477 |
args=config,
|
| 478 |
train_dataset=make_dataset(0),
|
| 479 |
reward_funcs=reward_fn,
|
| 480 |
+
processing_class=tokenizer,
|
| 481 |
)
|
| 482 |
|
| 483 |
# ββ Curriculum callback βββββββββββββββββββββββββββββββββββββββββββββββββββββββ
|