9 Commits

Author	SHA1	Message	Date
Gnome Ann	5e3c7c07ae	Merge branch 'main' into neox	2022-06-21 19:30:51 -04:00
Gnome Ann	a7e3ef71aa	Add final layer norm to OPT	2022-06-21 16:36:26 -04:00
Gnome Ann	4fa5f1cd6a	Add TPU support for OPT-350M ... The 350M model seems to have a different structure than the other ones ???	2022-05-12 22:21:15 -04:00
Gnome Ann	f5e689a725	Upload maps/opt.json and update requirements	2022-05-12 19:09:31 -04:00
Gnome Ann	a61ba0d000	Upload map file for GPT-NeoX	2022-04-29 00:41:56 -04:00
Gnome Ann	2d38e90509	Remove lm_head.weight from maps/xglm.json	2022-04-20 12:56:57 -04:00
Gnome Ann	4625158d30	Fix typo in previous commit	2022-03-05 12:56:42 -05:00
Gnome Ann	0a258a6282	Support for loading HF models on TPU with --colab_tpu	2022-03-05 12:33:33 -05:00
Gnome Ann	58a2c18821	Add lazy torch loading support to transformers backend	2022-03-04 00:33:10 -05:00