gonzalolinares commited on
Commit
6831b86
·
verified ·
1 Parent(s): 625c0f0

fix GRPOConfig: remove invalid max_length

Browse files
Files changed (1) hide show
  1. train_grpo.py +1 -1
train_grpo.py CHANGED
@@ -167,9 +167,9 @@ def main() -> None:
167
  save_strategy="steps",
168
  save_steps=50,
169
  save_total_limit=1,
170
- max_length=1024,
171
  temperature=0.7,
172
  bf16=True,
 
173
  push_to_hub=False,
174
  hub_model_id=HUB_MODEL_ID,
175
  report_to="none",
 
167
  save_strategy="steps",
168
  save_steps=50,
169
  save_total_limit=1,
 
170
  temperature=0.7,
171
  bf16=True,
172
+ remove_unused_columns=False,
173
  push_to_hub=False,
174
  hub_model_id=HUB_MODEL_ID,
175
  report_to="none",