diff --git a/docs/source/reducing_memory_usage.md b/docs/source/reducing_memory_usage.md index ecc921168e..e015c43906 100644 --- a/docs/source/reducing_memory_usage.md +++ b/docs/source/reducing_memory_usage.md @@ -55,7 +55,7 @@ To set the truncation parameter, use the following code snippet: ```python from trl import SFTConfig -training_args = SFTConfig(..., max_length=...) +training_args = SFTConfig(..., max_seq_length=...) ```