15 Commits

Include renames

Author	SHA1	Message	Date
Adam Novak	28c812c832	Use all logical cores in Torch mode	2022-06-28 22:26:51 -04:00
Brett Kuprel	1fbb209623	fixed bug with cuda in detokenizer	2022-06-28 22:02:35 -04:00
Brett Kuprel	764b0bc685	cuda in detokenizer from previous commit broke colab flax model, fixed	2022-06-28 21:36:48 -04:00
Brett Kuprel	17c96fe110	works with cuda	2022-06-28 21:28:36 -04:00
Brett Kuprel	9d6b6dcc92	previous commit broke flax model, fixed now	2022-06-28 12:54:58 -04:00
Brett Kuprel	5aa6fe49bf	use cuda if available	2022-06-28 12:47:11 -04:00
Brett Kuprel	8544f59576	use cuda if available	2022-06-28 12:38:31 -04:00
Brett Kuprel	aef24ea157	torch.no_grad(), cleanup	2022-06-28 12:16:44 -04:00
Brett Kuprel	34df2b97df	previous commit broke colab example, so adjusting flax requirement to 0.4.2 for now	2022-06-28 08:04:08 -04:00
Brett Kuprel	38ebe54a38	works with latest flax version 0.5.2 now	2022-06-28 07:12:29 -04:00
Brett Kuprel	a014dccc05	fixed an issue with argument parser	2022-06-27 16:49:42 -04:00
Brett Kuprel	e7001f063c	simplified	2022-06-27 15:46:04 -04:00
Brett Kuprel	18e6a9852f	license and cleanup	2022-06-27 14:34:10 -04:00
Brett Kuprel	c936d26102	back to linear attention	2022-06-27 13:19:03 -04:00
Brett Kuprel	018414a5c3	fixed relative imports	2022-06-27 12:43:47 -04:00