tanner/min-dalle-test
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity
68 Commits 1 Branch 0 Tags
61cc99c13c3753cdbfc74e4089a2dd34e3578ad4
Commit Graph
12 Commits
Author SHA1 Message Date
Brett Kuprel 661ec976ac simplified attention for torch model 2022-06-29 13:48:12 -04:00
Brett Kuprel ed91ab4a30 refactored to load models once and run multiple times 2022-06-29 09:42:12 -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 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 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