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

Merge pull request #73 from YellowRoseCx/main

Browse Source 
Create requirements-rocm.txt
This commit is contained in:
Cohee
2023-07-18 21:36:27 +03:00
committed by GitHub
parent 0b25aff3e1 1fddd2b03a
commit 8fd3f917a1
1 changed files with 19 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
requirements-rocm.txt Normal file
View File

@@ -0,0 +1,19 @@
flask
flask-cloudflared
flask-cors
flask-compress
markdown
Pillow
colorama
webuiapi
--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
accelerate
transformers==4.28.1
diffusers==0.16.1
silero-api-server
chromadb
sentence_transformers
edge-tts