rixty
/
KoboldAI-Client
mirror of https://github.com/KoboldAI/KoboldAI-Client.git
1
1
Fork 0
Code Issues Projects Releases Wiki Activity

Add a requirements.txt for TPU backend

This commit is contained in:
Gnome Ann 2021-11-19 18:20:02 -05:00
parent a65c4de840
commit 286ed51534
1 changed files with 13 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
requirements_mtj.txt Normal file
View File

@ -0,0 +1,13 @@
numpy
tqdm
requests
optax
dm-haiku >= 0.0.5, <= 0.0.9
ray[default]
jax == 0.2.12
transformers
progressbar2
git+https://github.com/VE-FORBRYDERNE/mesh-transformer-jax@ck
flask
Flask-SocketIO
flask-cloudflared >= 0.0.5