tanner/min-dalle-test
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity
43 Commits 1 Branch 0 Tags
0af6b1731fee2ce58cc485405c0c6ad98d763c3c
Commit Graph
12 Commits
Author SHA1 Message Date
Haydn Jones a3a247e6ec Fixed disabling of gradients in the torch code 2022-06-28 17:57:44 -06: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