tanner/min-dalle-test
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity
49 Commits 1 Branch 0 Tags
28c812c8326c0f9366fe3223731bc3bbcc6759ce
Commit Graph
15 Commits
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