7bf6c9a23f
Remove TPU Colab's dependency on optax and chex
2022-09-15 13:47:48 -04:00
551565c5ac
Fix error in aria2_hook when transformers version is at least 4.22.0
...
Some of the transformers.file_utils functions that were removed in
transformers v4.22.0 have equivalents in the huggingface_hub module.
2022-09-15 13:37:50 -04:00
aac999c073
Pin transformers to 4.21.3
2022-09-15 00:25:33 +02:00
359f95aff1
Downgrade Chex
...
Chex got a breaking update, while it is not a direct dependency we add it so that optax and jax don't break.
2022-09-13 14:59:23 +02:00
70aa182671
Restrict jaxlib version in TPU Colabs
2022-07-12 16:30:26 -04:00
b1d8797a54
Allow TPU Colab to load sharded HF models
2022-05-12 23:51:40 -04:00
f5e689a725
Upload maps/opt.json and update requirements
2022-05-12 19:09:31 -04:00
9a2d346d60
Merge branch 'main' into united
2022-04-12 10:41:51 +02:00
b377887336
Cap Bleach version to 4.1.0
2022-04-12 10:32:27 +02:00
66bc7e10bf
Remove Ray from requirements_mtj.txt
...
I made some changes recently to mesh transformer JAX so that we don't
need Ray anymore. This should make the installation a little faster.
2022-04-04 12:42:33 -04:00
f581fe89cb
Torch version changes
2022-03-17 21:11:36 +01:00
eaf190469d
Add PyTorch 1.11 support for lazy loader
2022-03-17 12:51:41 -04:00
9235754eb9
Dependency Fixes
2022-03-17 00:35:59 +01:00
50a96485a9
Fix dm-haiku
...
They did a change that breaks compatibility with our other dependencies, forcing version 0.0.5 to fix this.
2022-02-14 18:05:50 +01:00
a484244392
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
18ddd77337
Apply VE's changes to colab
2021-12-14 02:13:43 +01:00
bc1c0c4fa7
pip requirements
...
For Colab
2021-12-09 23:50:21 +01:00
3c349e6aaf
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 10:13:02 -05:00
4244b588cb
Ngrok requirements
...
Adds ngrok to the requirements.txt files
2021-11-29 18:13:30 +01:00
68e4b66fc5
Fix a typo in requirements_mtj.txt
2021-11-19 22:28:34 -05:00
286ed51534
Add a requirements.txt for TPU backend
2021-11-19 18:20:02 -05:00
vfbd
Henk
Gnome Ann
henk717