Skip to content

Commit

Permalink
Change alpaca_dataset train_on_input doc to match default value (pyto…
Browse files Browse the repository at this point in the history
  • Loading branch information
mirceamironenco authored Jan 3, 2025
1 parent d23fa93 commit 437c43e
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion torchtune/datasets/_alpaca.py
Original file line number Diff line number Diff line change
Expand Up @@ -51,7 +51,7 @@ def alpaca_dataset(
:class:`~torchtune.data.AlpacaToMessages` to the new column names in the dataset. Keys should be
"instruction", "input", and "output" and values should be the actual column names. If None, uses
the default column names ``"instruction``, ``"input"``, and ``"output"`` in ``tatsu-lab/alpaca``.
train_on_input (bool): Whether the model is trained on the prompt or not. Default is False.
train_on_input (bool): Whether the model is trained on the prompt or not. Default is True.
packed (bool): Whether or not to pack the dataset to ``max_seq_len`` prior to training. Default is False.
filter_fn (Optional[Callable]): callable used to filter the dataset prior to any pre-processing. See
the Hugging Face `docs <https://huggingface.co/docs/datasets/v2.20.0/process#select-and-filter>`_ for more
Expand Down

0 comments on commit 437c43e

Please sign in to comment.