KoboldAI-Client/requirements_mtj.txt

torch >= 1.9, < 1.13
numpy
tqdm
requests
dm-haiku==0.0.9
jax==0.3.25
jaxlib==0.3.25
chex == 0.1.5
transformers == 4.24.0
huggingface_hub==0.12.1
progressbar2
git+https://github.com/VE-FORBRYDERNE/mesh-transformer-jax@ck
Flask==2.2.3
Flask-SocketIO==5.3.2
python-socketio==5.7.2
flask-cloudflared==0.0.10
flask-ngrok
eventlet==0.33.3
dnspython==2.2.1
lupa==1.10
markdown
bleach==4.1.0
flask-session==0.4.0
marshmallow>=3.13
apispec-webframeworks
loguru
Torch fixes for colab 2022-09-25 19:47:08 +02:00			`torch >= 1.9, < 1.13`
Add a requirements.txt for TPU backend 2021-11-20 00:20:02 +01:00			`numpy`
			`tqdm`
			`requests`
Fix: TPU driver error to_dlpack/from_dlpack was causing issues with tensor with new jax version 2023-04-23 01:49:42 +02:00			`dm-haiku==0.0.9`
			`jax==0.3.25`
			`jaxlib==0.3.25`
Fix the chex dependency 2023-03-30 23:41:35 +02:00			`chex == 0.1.5`
Pin transformers version To avoid breaking changes lets force the exact transformers version we code against. This will be automatically picked up by all the automatic updaters. 2022-12-02 01:48:12 +01:00			`transformers == 4.24.0`
Huggingface Hub Pin 2023-03-08 18:03:36 +01:00			`huggingface_hub==0.12.1`
Add a requirements.txt for TPU backend 2021-11-20 00:20:02 +01:00			`progressbar2`
Allow TPU Colab to load sharded HF models 2022-05-13 05:51:40 +02:00			`git+https://github.com/VE-FORBRYDERNE/mesh-transformer-jax@ck`
SocketIO Requirements Pin 2023-03-17 01:28:59 +01:00			`Flask==2.2.3`
			`Flask-SocketIO==5.3.2`
			`python-socketio==5.7.2`
Pin Flask-cloudflared 2023-02-10 19:11:13 +01:00			`flask-cloudflared==0.0.10`
Modify TPU backend code to support JAX 0.2.21 The original one supported versions of JAX up to 0.2.12, and possibly also some earlier versions. This new code supports exclusively JAX 0.2.21 and does not work with any earlier or later versions of JAX. However, this new code benefits from not needing to recompile when changing "Amount To Generate" and also from supporting stopping generation early, which makes an implementation of Dynamic World Info Scan finally possible. 2021-11-30 16:13:02 +01:00			`flask-ngrok`
SocketIO Requirements Pin 2023-03-17 01:28:59 +01:00			`eventlet==0.33.3`
DNSPython workaround DNSPython had an update eventlet is not ready for. We now manually cap DNSPython to ensure the installations still happen correctly. 2023-01-16 16:32:17 +01:00			`dnspython==2.2.1`
Welcome Message API Allows model creators to customize the welcome message using Markdown and Limited HTML Existing United users need to run install_requirements..bat again, you can leave the existing dependencies intact. 2022-01-30 19:47:30 +01:00			`lupa==1.10`
			`markdown`
Cap Bleach version to 4.1.0 2022-04-12 10:32:27 +02:00			`bleach==4.1.0`
SocketIO Requirements Pin 2023-03-17 01:28:59 +01:00			`flask-session==0.4.0`
Upload basic API with /generate POST endpoint 2022-08-08 08:27:48 +02:00			`marshmallow>=3.13`
			`apispec-webframeworks`
Requirements Updates 2022-09-25 19:33:18 +02:00			`loguru`