diff --git a/README.md b/README.md index 1cb5fe9be..1d1965e2c 100644 --- a/README.md +++ b/README.md @@ -124,9 +124,9 @@ pip install unsloth #### Notes To run Unsloth directly on Windows: - Install Triton from this Windows fork and follow the instructions [here](https://github.com/woct0rdho/triton-windows) (be aware that the Windows fork requires PyTorch >= 2.4 and CUDA 12) -- In the SFTTrainer, set `dataset_num_proc=1` to avoid a crashing issue: +- In the `SFTConfig`, set `dataset_num_proc=1` to avoid a crashing issue: ```python -trainer = SFTTrainer( +SFTConfig( dataset_num_proc=1, ... )