ostris/ai-toolkit
1
0
Fork 0
mirror of https://github.com/ostris/ai-toolkit.git synced 2026-01-26 16:39:47 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update readme for torch requirements

Browse Source
This commit is contained in:
Jaret Burkett
2023-11-28 12:52:55 -07:00
parent be815f9c47
commit 0d5943af91
2 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
README.md
View File

@@ -35,8 +35,8 @@ git submodule update --init --recursive
python3 -m venv venv
source venv/bin/activate
# .\venv\Scripts\activate on windows
# windows install pytorch first with
# pip3 install torch torchvision --index-url https://download.pytorch.org/whl/cu117
# install torch first
pip3 install torch
pip3 install -r requirements.txt
```

1
requirements.txt
View File

@@ -23,3 +23,4 @@ prodigyopt
controlnet_aux==0.0.7
python-dotenv
bitsandbytes
xformers