Skip to content

Commit bbd8cbb

Browse files
authored
🤐 Fix deprecation warnings (huggingface#2395)
1 parent 4f937c7 commit bbd8cbb

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

trl/trainer/ppo_trainer.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -96,8 +96,10 @@ def forward(self, **kwargs):
9696
class PPOTrainer(Trainer):
9797
_tag_names = ["trl", "ppo"]
9898

99-
@deprecate_kwarg("config", new_name="args", version="0.15.0", raise_if_both_names=True)
100-
@deprecate_kwarg("tokenizer", new_name="processing_class", version="0.15.0", raise_if_both_names=True)
99+
@deprecate_kwarg("config", "0.15.0", "args", warn_if_greater_or_equal_version=True, raise_if_both_names=True)
100+
@deprecate_kwarg(
101+
"tokenizer", "0.15.0", "processing_class", warn_if_greater_or_equal_version=True, raise_if_both_names=True
102+
)
101103
@deprecate_kwarg("policy", "0.15.0", "model", warn_if_greater_or_equal_version=True, raise_if_both_names=True)
102104
@deprecate_kwarg(
103105
"ref_policy", "0.15.0", "ref_model", warn_if_greater_or_equal_version=True, raise_if_both_names=True

0 commit comments

Comments
 (0)