Pass train_dataset (required by AsyncGRPOTrainer ctor in 1.15.0)
Browse files- 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,
|