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