Skip to content

Commit

Permalink
Update DPO EADME.md (#12162)
Browse files Browse the repository at this point in the history
bitsanbytes multi backend is now available and is required , otherwise would error out saying that no cuda is available
  • Loading branch information
rahulunair authored Oct 31, 2024
1 parent 29400e2 commit 4cf1ccc
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion python/llm/example/GPU/LLM-Finetuning/DPO/README.md
Original file line number Diff line number Diff line change
Expand Up @@ -19,7 +19,8 @@ conda activate llm
pip install --pre --upgrade ipex-llm[xpu] --extra-index-url https://pytorch-extension.intel.com/release-whl/stable/xpu/us/
pip install transformers==4.36.0 datasets
pip install trl peft==0.10.0
pip install bitsandbytes
# Note, if you don't want to reinstall BNBs dependencies, append the `--no-deps` flag!
pip install --no-deps --force-reinstall 'https://github.com/bitsandbytes-foundation/bitsandbytes/releases/download/continuous-release_multi-backend-refactor/bitsandbytes-0.44.1.dev0-py3-none-manylinux_2_24_x86_64.whl'
```

### 2. Configures OneAPI environment variables
Expand Down

0 comments on commit 4cf1ccc

Please sign in to comment.