fixed bug I added to demo config

2024-08-11 10:28:39 -06:00
parent ec1ea7aa0e
commit 8d48ad4e85
2 changed files with 1 additions and 2 deletions
--- a/README.md
+++ b/README.md
@@ -39,7 +39,7 @@ cd ai-toolkit
 git submodule update --init --recursive
 python -m venv venv
 .\venv\Scripts\activate
-pip install torch --use-pep517 --extra-index-url https://download.pytorch.org/whl/cu118
+pip install torch torchvision --index-url https://download.pytorch.org/whl/cu121
 pip install -r requirements.txt
 ```