Remove max_prompt_length (not in TRL 1.9 DPOConfig)
Browse files- train_dpo.py +0 -1
train_dpo.py
CHANGED
|
@@ -201,7 +201,6 @@ def train_dpo(config: Config):
|
|
| 201 |
dataloader_num_workers=2,
|
| 202 |
beta=config.beta,
|
| 203 |
max_length=config.max_length,
|
| 204 |
-
max_prompt_length=config.max_prompt_length,
|
| 205 |
)
|
| 206 |
|
| 207 |
# DPO Trainer
|
|
|
|
| 201 |
dataloader_num_workers=2,
|
| 202 |
beta=config.beta,
|
| 203 |
max_length=config.max_length,
|
|
|
|
| 204 |
)
|
| 205 |
|
| 206 |
# DPO Trainer
|