fix DPOConfig: drop max_prompt_length
Browse files- train_dpo.py +0 -1
train_dpo.py
CHANGED
|
@@ -104,7 +104,6 @@ def main() -> None:
|
|
| 104 |
save_strategy="epoch",
|
| 105 |
save_total_limit=1,
|
| 106 |
max_length=1024,
|
| 107 |
-
max_prompt_length=512,
|
| 108 |
bf16=True,
|
| 109 |
push_to_hub=False,
|
| 110 |
hub_model_id=HUB_MODEL_ID,
|
|
|
|
| 104 |
save_strategy="epoch",
|
| 105 |
save_total_limit=1,
|
| 106 |
max_length=1024,
|
|
|
|
| 107 |
bf16=True,
|
| 108 |
push_to_hub=False,
|
| 109 |
hub_model_id=HUB_MODEL_ID,
|