From a802014ec588511dbc4c50acca3ce5fc5b572ba8 Mon Sep 17 00:00:00 2001 From: Jaret Burkett Date: Thu, 26 Mar 2026 12:15:32 -0600 Subject: [PATCH] Update the torch versions in the README --- README.md | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/README.md b/README.md index fdc54fad..96ac86d5 100644 --- a/README.md +++ b/README.md @@ -249,7 +249,7 @@ cd ai-toolkit python3 -m venv venv source venv/bin/activate # install torch first -pip3 install --no-cache-dir torch==2.7.0 torchvision==0.22.0 torchaudio==2.7.0 --index-url https://download.pytorch.org/whl/cu126 +pip3 install --no-cache-dir torch==2.9.1 torchvision==0.24.1 torchaudio==2.9.1 --index-url https://download.pytorch.org/whl/cu128 pip3 install -r requirements.txt ``` @@ -265,7 +265,7 @@ git clone https://github.com/ostris/ai-toolkit.git cd ai-toolkit python -m venv venv .\venv\Scripts\activate -pip install --no-cache-dir torch==2.7.0 torchvision==0.22.0 torchaudio==2.7.0 --index-url https://download.pytorch.org/whl/cu126 +pip install --no-cache-dir torch==2.9.1 torchvision==0.24.1 torchaudio==2.9.1 --index-url https://download.pytorch.org/whl/cu128 pip install -r requirements.txt ```