tanner/min-dalle-test
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity
114 Commits 1 Branch 0 Tags
7bf76deafb87212da1312513968ee901a1a40bc7
Commit Graph
82 Commits
This Branch
This Branch
All Branches
Author SHA1 Message Date
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
Brett Kuprel 6a068651e5 updated colab 2022-06-27 20:59:39 -04:00
Brett Kuprel 24d8e29ef2 readme formatting 2022-06-27 17:28:40 -04:00
Brett Kuprel 5444c73e3f Merge branch 'main' of https://github.com/kuprel/min-dalle 
commited from colab separately
 2022-06-27 17:26:53 -04:00
Brett Kuprel 8363495f0a updated readme 2022-06-27 17:26:05 -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 32b7aa196b readme 2022-06-27 13:51:48 -04:00
Brett Kuprel 56b24494ae examples 2022-06-27 13:39:00 -04:00
Brett Kuprel 194ae7dfa1 examples 2022-06-27 13:38:35 -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
Brett Kuprel 9cf9d46dd7 first commit 2022-06-27 11:57:56 -04:00
Brett Kuprel 97fe8515f1 first commit 2022-06-27 11:55:55 -04:00