Unverified Commit a48310de authored by Pasquale Minervini's avatar Pasquale Minervini Committed by GitHub
Browse files

Update training_args.py -- a nightly install is not required anymore for torch.compile (#22266)

Update training_args.py

A nightly install is not required anymore for `torch.compile`.
parent 60d51ef5
...@@ -557,7 +557,7 @@ class TrainingArguments: ...@@ -557,7 +557,7 @@ class TrainingArguments:
Whether to use Apple Silicon chip based `mps` device. Whether to use Apple Silicon chip based `mps` device.
torch_compile (`bool`, *optional*, defaults to `False`): torch_compile (`bool`, *optional*, defaults to `False`):
Whether or not to compile the model using PyTorch 2.0 Whether or not to compile the model using PyTorch 2.0
[`torch.compile`](https://pytorch.org/get-started/pytorch-2.0/) (requires a nighlty install of PyTorch). [`torch.compile`](https://pytorch.org/get-started/pytorch-2.0/).
This will use the best defaults for the [`torch.compile` This will use the best defaults for the [`torch.compile`
API](https://pytorch.org/docs/2.0/generated/torch.compile.html?highlight=torch+compile#torch.compile). You API](https://pytorch.org/docs/2.0/generated/torch.compile.html?highlight=torch+compile#torch.compile). You
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment