diff --git a/README.md b/README.md index 680206393..0c25be6d4 100644 --- a/README.md +++ b/README.md @@ -133,6 +133,12 @@ python -m venv venv pip install --no-cache-dir torch==2.7.0 torchvision==0.22.0 --index-url https://download.pytorch.org/whl/cu126 pip install -r requirements.txt ``` +# If you use NVIDIA RTX 50-series +you should pip this version +``` +pip install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu128 +pip install -U bitsandbytes +``` # AI Toolkit UI @@ -451,4 +457,4 @@ Only larger updates are listed here. There are usually smaller daily updated tha - Added support for SDXL in the UI - Added support for SD 1.5 in the UI - Fixed UI Wan 2.1 14b name bug -- Added support for for conv training in the UI for models that support it \ No newline at end of file +- Added support for for conv training in the UI for models that support it