tobil commited on
Commit
c27b1da
·
verified ·
1 Parent(s): 4ffd0d6

Upload train_grpo.py with huggingface_hub

Browse files
Files changed (1) hide show
  1. train_grpo.py +1 -1
train_grpo.py CHANGED
@@ -269,7 +269,7 @@ def main():
269
  print("Initializing GRPO trainer...")
270
  trainer = GRPOTrainer(
271
  model=model,
272
- tokenizer=tokenizer,
273
  config=config,
274
  train_dataset=dataset,
275
  reward_funcs=reward_fn,
 
269
  print("Initializing GRPO trainer...")
270
  trainer = GRPOTrainer(
271
  model=model,
272
+ processing_class=tokenizer,
273
  config=config,
274
  train_dataset=dataset,
275
  reward_funcs=reward_fn,