fix GRPOConfig: remove invalid max_length
Browse files- 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",
|