Commit Graph
100 Commits
Author SHA1 Message Date
Brett Kuprel 849e720313 add default arguments to replicate 2022-07-02 10:46:56 -04:00
Brett Kuprel 465b2428de update replicate files 2022-07-02 10:35:27 -04:00
Brett Kuprel 9cde5b1a60 update replicate files 2022-07-02 10:32:17 -04:00
Brett Kuprel e6f1c3ef4c update replicate files 2022-07-02 10:05:16 -04:00
Brett KuprelandGitHub a91a259293 Merge pull request #49 from raphant/patch-1
Update README.md to include all required third-party dependencies
2022-07-02 09:32:25 -04:00
Brett KuprelandGitHub 8b77428102 Merge branch 'main' into patch-1 2022-07-02 09:32:15 -04:00
Brett Kuprel f8cae82b93 update readme 2022-07-02 09:26:30 -04:00
Brett Kuprel 20a91e8714 add grid-size example to readme 2022-07-02 09:23:59 -04:00
Brett Kuprel 72cd1501f2 update replicate file 2022-07-02 09:20:42 -04:00
Brett Kuprel e087c53482 update readme 2022-07-02 09:13:05 -04:00
Brett Kuprel 313635e914 fixed typing error for older python versions 2022-07-02 09:06:22 -04:00
Brett Kuprel 1eb56737d8 added grid_size parameter to generate a grid of images 2022-07-02 08:45:49 -04:00
Brett Kuprel e0386f991c removed typing_extensions requirement 2022-07-02 06:34:47 -04:00
Brett Kuprel 4b632555b5 models_root command line argument 2022-07-01 23:39:48 -04:00
Brett Kuprel 21c633b56e update readme 2022-07-01 20:27:29 -04:00
Brett Kuprel 9cbaa83b05 update readme 2022-07-01 20:25:49 -04:00
Brett Kuprel 2c9d19e1b7 update readme 2022-07-01 20:24:19 -04:00
Brett Kuprel cf9656baa2 added is_verbose flag 2022-07-01 20:17:20 -04:00
Brett Kuprel 35e97768a5 v0.2.0, MinDalleTorch -> MinDalle, breaking change 2022-07-01 19:44:24 -04:00
Brett Kuprel 2080e596c3 update readme 2022-07-01 19:19:02 -04:00
Brett Kuprel b3cd959061 update readme 2022-07-01 19:15:31 -04:00
Brett Kuprel e4c2b71c11 update readme 2022-07-01 19:12:43 -04:00
Brett Kuprel cc9bdb8f7b update readme 2022-07-01 19:08:42 -04:00
Brett Kuprel 847daa5be2 update readme 2022-07-01 19:05:46 -04:00
Brett Kuprel fb77728ea8 update readme 2022-07-01 19:03:18 -04:00
Brett Kuprel dc0d02a7a7 update readme 2022-07-01 19:02:49 -04:00
Brett Kuprel 2614a83f35 update readme 2022-07-01 19:01:07 -04:00
Brett Kuprel 8f58a51a7d fix readme 2022-07-01 18:54:05 -04:00
Brett Kuprel f1f433048c fix readme 2022-07-01 18:52:40 -04:00
Brett Kuprel f86a4595e8 fix readme 2022-07-01 18:51:06 -04:00
Brett Kuprel fff44d683e update readme, random seed if none specified 2022-07-01 18:50:11 -04:00
Brett Kuprel ec9e22e4d4 code format 2022-07-01 18:32:48 -04:00
Brett Kuprel 7b5d972914 update readme 2022-07-01 18:30:22 -04:00
Brett Kuprel be2beca7c0 added to pypi 2022-07-01 18:16:55 -04:00
Brett Kuprel 69683fec94 simplify import 2022-07-01 16:02:05 -04:00
Brett Kuprel 18c72ed34d simplified MinDalleTorch 2022-07-01 15:53:39 -04:00
Brett KuprelandGitHub 56f0563ad1 Merge pull request #46 from w4ffl35/update-requirements-to-include-versions
Updates requirements with versions
2022-07-01 14:36:15 -04:00
Brett Kuprel 2097c6ed5a cleanup 2022-07-01 14:35:13 -04:00
Brett Kuprel 70bd3cb80d update examples 2022-07-01 14:28:51 -04:00
Brett Kuprel 73d49af07f update examples 2022-07-01 14:26:36 -04:00
Brett Kuprel 763018e212 fix colab setup 2022-07-01 14:23:22 -04:00
Brett Kuprel 867f7d6970 update readme 2022-07-01 14:19:35 -04:00
Brett Kuprel 07ce93d5f8 moved flax model and conversion code to separate repository 2022-07-01 14:06:50 -04:00
Brett Kuprel febd18df77 update readme 2022-07-01 12:55:30 -04:00
Brett Kuprel b61041f766 add back requirements.txt 2022-07-01 12:53:55 -04:00
Brett Kuprel f71bd054fd Merge branch 'main' of https://github.com/kuprel/min-dalle 2022-07-01 12:21:39 -04:00
Brett Kuprel ee846ba1cf update setup scripts 2022-07-01 12:21:31 -04:00
Brett Kuprel 85f5866eff remove config.json dependency, default to torch in image_from_text.py 2022-07-01 12:03:37 -04:00
Brett Kuprel 4404e70764 fixed setup_torch.sh 2022-07-01 11:38:11 -04:00
Brett Kuprel 5cc7d0038f fixed setup_flax.sh 2022-07-01 11:32:36 -04:00
Brett Kuprel 2a4e3db15b update setup_torch.sh to download mini model too 2022-07-01 11:27:26 -04:00
Brett Kuprel ff77e96b07 update readme 2022-07-01 11:19:04 -04:00
Brett Kuprel b785accffd update readme 2022-07-01 11:18:30 -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 KuprelandGitHub 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 KuprelandGitHub e3329a7f64 Merge branch 'main' into patch-1 2022-07-01 02:40:38 -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 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 b913b58353 pre converting params to torch allows mega to run in standard colab runtime 2022-06-30 14:54:08 -04:00
Brett KuprelandGitHub 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 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 KuprelandGitHub 38377107da Update setup.sh 2022-06-29 21:45:05 -04:00
Brett KuprelandGitHub c76226571e Update cog.yaml 2022-06-29 19:14:25 -04:00
Brett KuprelandGitHub 8b86152359 Update cog.yaml 2022-06-29 19:14:10 -04:00
Brett Kuprel b7b6df23f7 updated replicate predict.py file 2022-06-29 15:53:25 -04:00
Brett KuprelandGitHub 005ee4938e Update README.md 2022-06-29 15:24:09 -04:00
Brett KuprelandGitHub 9fbfe0b76e Delete predict.py 2022-06-29 15:20:45 -04:00
Brett KuprelandGitHub 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
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
Brett Kuprel a4df279fd2 simplified attention and keys_values state resulted in decrease in inference time to 7.3 seconds (from ~10 seconds) 2022-06-29 13:56:29 -04:00
Brett Kuprel 661ec976ac simplified attention for torch model 2022-06-29 13:48:12 -04:00
Brett Kuprel 95afa18893 add gitattributes file 2022-06-29 12:45:41 -04:00
Brett Kuprel 764b5bbc0e works with latest flax version 0.5.2, updated requirements.txt 2022-06-29 11:46:19 -04:00
Brett Kuprel c4f613c89f readme wording 2022-06-29 11:01:46 -04:00
Brett Kuprel 6046863805 readme wording 2022-06-29 11:00:38 -04:00
Brett Kuprel 2b552fe9db readme wording 2022-06-29 10:54:01 -04:00
Brett Kuprel b7c2414c76 readme wording 2022-06-29 10:47:08 -04:00
Brett Kuprel 4e62f85ab9 readme wording 2022-06-29 10:45:46 -04:00
Brett Kuprel 53695e32f7 updated readme with torch examples 2022-06-29 10:43:46 -04:00