Commit Graph

317 Commits

Author SHA1 Message Date
Brett Kuprel b785accffd update readme 2022-07-01 11:18:30 -04:00
Brett Kuprel cc90fa4a90 download pre-converted torch files in colab 2022-07-01 11:09:49 -04:00
Brett Kuprel 09a0f85b8e separate setup processes for flax and torch 2022-07-01 11:08:33 -04:00
Brett Kuprel 7bf76deafb fixed wrong file path 2022-07-01 10:58:29 -04:00
Brett Kuprel fffd0f2b83 Merge branch 'main' of https://github.com/kuprel/min-dalle 2022-07-01 10:17:40 -04:00
Brett Kuprel e4c2be54cb save converted detokenizer params 2022-07-01 10:17:29 -04:00
Brett Kuprel 26c336bd4d Merge pull request #20 from Ewpratten/patch-1
Explicitly call wandb as a python module.
2022-07-01 07:55:48 -04:00
Brett Kuprel 8b5960b687 update setup.sh 2022-07-01 07:54:19 -04:00
Brett Kuprel d13e573fb8 mega works with latest flax version 0.5.2 now, removing 0.4.2 pin 2022-07-01 03:07:32 -04:00
Brett Kuprel b40fd83a0d mega works with latest flax version 0.5.2 now, removing 0.4.2 pin 2022-07-01 02:58:43 -04:00
Brett Kuprel e3329a7f64 Merge branch 'main' into patch-1 2022-07-01 02:40:38 -04:00
Brett Kuprel 9ac9c0ca30 simplified colab 2022-06-30 21:23:15 -04:00
Brett Kuprel eaee59a1ef update readme 2022-06-30 21:05:02 -04:00
Brett Kuprel e9ff397009 Merge branch 'main' of https://github.com/kuprel/min-dalle 2022-06-30 21:04:06 -04:00
Brett Kuprel f8bffc6892 update readme 2022-06-30 21:03:56 -04:00
Brett Kuprel b93c4fd0e4 simplified colab 2022-06-30 21:02:27 -04:00
Brett Kuprel 432ffa8d8c reusable mega torch model works in standard colab runtime 2022-06-30 20:55:42 -04:00
Brett Kuprel 08b158d580 updated readme 2022-06-30 16:50:04 -04:00
Brett Kuprel 2311a1af7b delete cache 2022-06-30 15:48:20 -04:00
Brett Kuprel d9d7f34b22 update readme 2022-06-30 15:33:53 -04:00
Brett Kuprel 23620b95cd Merge branch 'main' of https://github.com/kuprel/min-dalle 2022-06-30 15:19:21 -04:00
Brett Kuprel c8f3304363 update readme 2022-06-30 15:18:29 -04:00
Brett Kuprel a3d81888b5 generate avocado armchair with pytorch in standard colab runtime 2022-06-30 15:17:35 -04:00
Brett Kuprel ed0193296c mega model works in standard colab runtime 2022-06-30 14:57:53 -04:00
Brett Kuprel b913b58353 pre converting params to torch allows mega to run in standard colab runtime 2022-06-30 14:54:08 -04:00
Brett Kuprel de97fcf06b Update requirements.txt 2022-06-30 13:13:33 -04:00
Brett Kuprel c2a3858c96 delete params sooner 2022-06-30 11:44:36 -04:00
Brett Kuprel f951424e38 is_reusable 2022-06-30 11:25:24 -04:00
Brett Kuprel 3e64e868ef Merge branch 'main' of https://github.com/kuprel/min-dalle 2022-06-30 11:09:17 -04:00
Brett Kuprel b55bcba4c0 removed deepcopy, delete expendable parameters after use 2022-06-30 11:09:09 -04:00
Brett Kuprel b44e8997a0 still some issues with mega in standard runtime, reverting for now 2022-06-30 10:02:08 -04:00
Brett Kuprel 6d1ac07442 mega model in standard runtime 2022-06-30 09:43:13 -04:00
Brett Kuprel e9c01e32a5 mega model works with standard colab runtime in expendable mode 2022-06-30 09:39:14 -04:00
Brett Kuprel 41a44068d0 keep params in expendable mode 2022-06-30 09:36:32 -04:00
Brett Kuprel df9aa6f915 sort -> topk, prev_token_and_index -> prev_token, token_index 2022-06-30 09:04:11 -04:00
Brett Kuprel fb97ba5e20 update readme, cleanup 2022-06-30 07:41:31 -04:00
Brett Kuprel 1e18ba0ffa is_expendable argument reduces memory usage for command line script 2022-06-30 06:43:10 -04:00
Brett Kuprel 38377107da Update setup.sh 2022-06-29 21:45:05 -04:00
Brett Kuprel c76226571e Update cog.yaml 2022-06-29 19:14:25 -04:00
Brett Kuprel 8b86152359 Update cog.yaml 2022-06-29 19:14:10 -04:00
Brett Kuprel 5f931f6899 simplified colab setup 2022-06-29 16:32:51 -04:00
Brett Kuprel b7b6df23f7 updated replicate predict.py file 2022-06-29 15:53:25 -04:00
Brett Kuprel 005ee4938e Update README.md 2022-06-29 15:24:09 -04:00
Brett Kuprel 9fbfe0b76e Delete predict.py 2022-06-29 15:20:45 -04:00
Brett Kuprel 15b0c03485 Merge pull request #38 from chenxwh/replicate
Add Web Demo & Docker environment
2022-06-29 15:20:20 -04:00
Brett Kuprel d99828a239 simplified flax attention and matched torch attention 2022-06-29 14:56:28 -04:00
Chenxi fcc17c895d Merge branch 'kuprel:main' into replicate 2022-06-29 19:50:47 +01:00
Chenxi eb9f4c6b3b replicate demo 2022-06-29 19:50:10 +01:00
Brett Kuprel 61cc99c13c read tokenizer files with utf8 encoding 2022-06-29 14:18:23 -04:00
Brett Kuprel 742d3609b0 Merge branch 'main' of https://github.com/kuprel/min-dalle 2022-06-29 13:56:39 -04:00