shank commited on
Commit
cb09ef1
Β·
1 Parent(s): 4668456

Fix: Fixed file

Browse files
Files changed (1) hide show
  1. 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
- tokenizer=tokenizer,
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 ───────────────────────────────────────────────────────