Commit Graph

16 Commits

Author SHA1 Message Date
Gnome Ann b1d8797a54 Allow TPU Colab to load sharded HF models 2022-05-12 23:51:40 -04:00
Gnome Ann f5e689a725 Upload maps/opt.json and update requirements 2022-05-12 19:09:31 -04:00
Henk 9a2d346d60 Merge branch 'main' into united 2022-04-12 10:41:51 +02:00
Henk b377887336 Cap Bleach version to 4.1.0 2022-04-12 10:32:27 +02:00
Gnome Ann 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
henk717 f581fe89cb Torch version changes 2022-03-17 21:11:36 +01:00
Gnome Ann eaf190469d Add PyTorch 1.11 support for lazy loader 2022-03-17 12:51:41 -04:00
henk717 9235754eb9 Dependency Fixes 2022-03-17 00:35:59 +01:00
henk717 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
henk717 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
henk717 18ddd77337 Apply VE's changes to colab 2021-12-14 02:13:43 +01:00
henk717 bc1c0c4fa7 pip requirements
For Colab
2021-12-09 23:50:21 +01:00
Gnome Ann 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
henk717 4244b588cb Ngrok requirements
Adds ngrok to the requirements.txt files
2021-11-29 18:13:30 +01:00
Gnome Ann 68e4b66fc5 Fix a typo in requirements_mtj.txt 2021-11-19 22:28:34 -05:00
Gnome Ann 286ed51534 Add a requirements.txt for TPU backend 2021-11-19 18:20:02 -05:00