SillyTavern/SillyTavern-extras
1
0
Fork 0
mirror of https://github.com/SillyTavern/SillyTavern-Extras.git synced 2026-04-28 10:21:20 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update pytorch version in requirements-rocm.txt

Browse Source
This commit is contained in:
YellowRoseCx
2023-10-24 19:46:40 -05:00
committed by GitHub
parent 2830a5ff4e
commit baf4f93c78
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
requirements-rocm.txt
View File

@@ -6,10 +6,10 @@ markdown==3.4.3
Pillow==9.5.0
colorama==0.4.6
webuiapi==0.9.5
--extra-index-url https://download.pytorch.org/whl/rocm5.4.2
torch>=2.0.0+rocm5.4.2,<2.1.0+rocm5.4.2
torchvision>=0.15.0+rocm5.4.2,<0.16.0+rocm5.4.2
torchaudio>=2.0.0+rocm5.4.2,<2.1.0+rocm5.4.2
--extra-index-url https://download.pytorch.org/whl/rocm5.6
torch>=2.0.0+rocm5.4.2,<=2.1.0+rocm5.6
torchvision>=0.15.0+rocm5.4.2,<=0.16.0+rocm5.6
torchaudio>=2.0.0+rocm5.4.2,<=2.1.0+rocm5.6
accelerate==0.20.3
transformers==4.28.1
diffusers==0.17.1