SaylorTwift HF Staff commited on
Commit
2a4d2a5
·
verified ·
1 Parent(s): 8067c88

Pass train_dataset (required by AsyncGRPOTrainer ctor in 1.15.0)

Browse files
Files changed (1) hide show
  1. train_lora.py +1 -0
train_lora.py CHANGED
@@ -138,6 +138,7 @@ def main():
138
  trainer = AsyncGRPOTrainer(
139
  model=MODEL,
140
  args=config,
 
141
  processing_class=tokenizer,
142
  rollout_worker=worker,
143
  peft_config=peft_config,
 
138
  trainer = AsyncGRPOTrainer(
139
  model=MODEL,
140
  args=config,
141
+ train_dataset=dataset, # required by the ctor even with an injected worker
142
  processing_class=tokenizer,
143
  rollout_worker=worker,
144
  peft_config=peft_config,