Skip to content

Commit c8a5add

Browse files
Fix PEFT interlinks in docstrings (#4178)
1 parent a7b54f9 commit c8a5add

File tree

3 files changed

+3
-3
lines changed

3 files changed

+3
-3
lines changed

docs/source/sft_trainer.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -201,7 +201,7 @@ trainer = SFTTrainer(
201201
trainer.train()
202202
```
203203

204-
You can also continue training your [`peft.PeftModel`]. For that, first load a `PeftModel` outside [`SFTTrainer`] and pass it directly to the trainer without the `peft_config` argument being passed.
204+
You can also continue training your [`~peft.PeftModel`]. For that, first load a `PeftModel` outside [`SFTTrainer`] and pass it directly to the trainer without the `peft_config` argument being passed.
205205

206206
```python
207207
from datasets import load_dataset

trl/trainer/gkd_trainer.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -85,7 +85,7 @@ class GKDTrainer(SFTTrainer):
8585
preprocess_logits_for_metrics (`Callable`, *optional*):
8686
Function to preprocess the logits before computing the metrics. Must take in the `logits` and `labels` and
8787
return the logits to be used for metrics computation.
88-
peft_config ([`~peft.config.PeftConfig`], *optional*):
88+
peft_config ([`~peft.PeftConfig`], *optional*):
8989
PEFT configuration to use PEFT for training. If `None`, PEFT is not used. If provided, the `model` will be
9090
wrapped with the specified PEFT adapter.
9191
formatting_func (`Callable`, *optional*):

trl/trainer/ppo_trainer.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -123,7 +123,7 @@ class PPOTrainer(BaseTrainer):
123123
[`~transformers.Trainer.create_optimizer_and_scheduler`] method.
124124
callbacks (`list` of [`~transformers.TrainerCallback`], *optional*):
125125
Callbacks to use during training.
126-
peft_config ([`~peft.config.PeftConfig`], *optional*):
126+
peft_config ([`~peft.PeftConfig`], *optional*):
127127
PEFT configuration to use PEFT for training. If `None`, PEFT is not used. If provided, the policy `model`
128128
will be wrapped with the specified PEFT adapter.
129129
"""

0 commit comments

Comments
 (0)