ostris/ai-toolkit
1
0
Fork 0
mirror of https://github.com/ostris/ai-toolkit.git synced 2026-05-01 03:31:35 +00:00
Code Issues Packages Projects Releases Wiki Activity

Bumped torch and cuda to support blackwell arch

Browse Source
This commit is contained in:
Jaret Burkett
2025-05-09 11:17:58 -06:00
parent 66a41e49d9
commit 64130189ce
4 changed files with 11 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
requirements.txt
View File

@@ -1,6 +1,6 @@
torch==2.6.0
torchvision==0.21.0
torchao==0.9.0
torch==2.7.0
torchvision==0.22.0
torchao==0.10.0
safetensors
git+https://github.com/jaretburkett/easy_dwpose.git
git+https://github.com/huggingface/diffusers@363d1ab7e24c5ed6c190abb00df66d9edb74383b