tanner/min-dalle-test
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity
151 Commits 1 Branch 0 Tags
cc9bdb8f7b19b02903ef0d56d341fa0f6e555010
Commit Graph
100 Commits
This Branch
This Branch
All Branches
Author SHA1 Message Date
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
Brett Kuprel ed91ab4a30 refactored to load models once and run multiple times 2022-06-29 09:42:12 -04:00
kuprelandGitHub 0fe3e2d2b9 Merge pull request #28 from interfect/torch-logical-cores 
Use all logical cores in Torch mode
 2022-06-28 22:57:02 -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 d846cab1b6 Merge branch 'main' of https://github.com/kuprel/min-dalle 2022-06-28 21:28:50 -04:00
Brett Kuprel 17c96fe110 works with cuda 2022-06-28 21:28:36 -04:00
kuprelandGitHub 0af6b1731f Update setup.sh 
removed `git lfs` dependency
 2022-06-28 21:18:24 -04:00
kuprelandGitHub 6f0be73579 Merge pull request #23 from andrewginns/main 
Simplified requirements
 2022-06-28 21:16:00 -04:00
kuprelandGitHub b8c4173181 Merge pull request #24 from TheFutureGadgetsLab/main 
Fixed disabling of gradients in the torch code
 2022-06-28 21:10:35 -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
kuprelandGitHub 1e2649148b Merge pull request #19 from jedahan/patch-2 
Support pip/pip3 binary names for setup
 2022-06-28 11:35:20 -04:00
kuprelandGitHub 57127ac616 Merge branch 'main' into patch-2 2022-06-28 11:34:46 -04:00
kuprelandGitHub 6baeeec7f1 Merge pull request #17 from jedahan/patch-1 
Exit script immediately if any command fails
 2022-06-28 11:31:42 -04:00
kuprelandGitHub 407ea96865 Merge pull request #14 from rupa/main 
fix requirements.txt
 2022-06-28 09:25:58 -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
kuprelandGitHub 50e1e74c4a Merge pull request #6 from osanseviero/patch-1 
Update README.md
 2022-06-28 06:28:21 -04:00
Brett Kuprel 6260252348 update readme 2022-06-27 22:08:29 -04:00
kuprelandGitHub 2ad7009a16 Update README.md 2022-06-27 21:11:27 -04:00