355 Commits (main)
 

Author SHA1 Message Date
Brett Kuprel eaee59a1ef update readme 2 years ago
Brett Kuprel e9ff397009 Merge branch 'main' of https://github.com/kuprel/min-dalle 2 years ago
Brett Kuprel f8bffc6892 update readme 2 years ago
Brett Kuprel b93c4fd0e4 simplified colab 2 years ago
Brett Kuprel 432ffa8d8c reusable mega torch model works in standard colab runtime 2 years ago
Brett Kuprel 08b158d580 updated readme 2 years ago
Brett Kuprel 2311a1af7b delete cache 2 years ago
Brett Kuprel d9d7f34b22 update readme 2 years ago
Brett Kuprel 23620b95cd Merge branch 'main' of https://github.com/kuprel/min-dalle 2 years ago
Brett Kuprel c8f3304363 update readme 2 years ago
Brett Kuprel a3d81888b5 generate avocado armchair with pytorch in standard colab runtime 2 years ago
Brett Kuprel ed0193296c mega model works in standard colab runtime 2 years ago
Brett Kuprel b913b58353 pre converting params to torch allows mega to run in standard colab runtime 2 years ago
Brett Kuprel de97fcf06b
Update requirements.txt 2 years ago
Brett Kuprel c2a3858c96 delete params sooner 2 years ago
Brett Kuprel f951424e38 is_reusable 2 years ago
Brett Kuprel 3e64e868ef Merge branch 'main' of https://github.com/kuprel/min-dalle 2 years ago
Brett Kuprel b55bcba4c0 removed deepcopy, delete expendable parameters after use 2 years ago
Brett Kuprel b44e8997a0 still some issues with mega in standard runtime, reverting for now 2 years ago
Brett Kuprel 6d1ac07442 mega model in standard runtime 2 years ago
Brett Kuprel e9c01e32a5 mega model works with standard colab runtime in expendable mode 2 years ago
Brett Kuprel 41a44068d0 keep params in expendable mode 2 years ago
Brett Kuprel df9aa6f915 sort -> topk, prev_token_and_index -> prev_token, token_index 2 years ago
Brett Kuprel fb97ba5e20 update readme, cleanup 2 years ago
Brett Kuprel 1e18ba0ffa is_expendable argument reduces memory usage for command line script 2 years ago
Brett Kuprel 38377107da
Update setup.sh 2 years ago
Brett Kuprel c76226571e
Update cog.yaml 2 years ago
Brett Kuprel 8b86152359
Update cog.yaml 2 years ago
Brett Kuprel 5f931f6899 simplified colab setup 2 years ago
Brett Kuprel b7b6df23f7 updated replicate predict.py file 2 years ago
Brett Kuprel 005ee4938e
Update README.md 2 years ago
Brett Kuprel 9fbfe0b76e
Delete predict.py 2 years ago
Brett Kuprel 15b0c03485
Merge pull request #38 from chenxwh/replicate 2 years ago
Brett Kuprel d99828a239 simplified flax attention and matched torch attention 2 years ago
Chenxi fcc17c895d
Merge branch 'kuprel:main' into replicate 2 years ago
Chenxi eb9f4c6b3b replicate demo 2 years ago
Brett Kuprel 61cc99c13c read tokenizer files with utf8 encoding 2 years ago
Brett Kuprel 742d3609b0 Merge branch 'main' of https://github.com/kuprel/min-dalle 2 years ago
Brett Kuprel a4df279fd2 simplified attention and keys_values state resulted in decrease in inference time to 7.3 seconds (from ~10 seconds) 2 years ago
Brett Kuprel 80558b8a82 time the inference pass 2 years ago
Brett Kuprel 661ec976ac simplified attention for torch model 2 years ago
Brett Kuprel 95afa18893 add gitattributes file 2 years ago
Brett Kuprel 764b5bbc0e works with latest flax version 0.5.2, updated requirements.txt 2 years ago
Brett Kuprel c4f613c89f readme wording 2 years ago
Brett Kuprel 6046863805 readme wording 2 years ago
Brett Kuprel 2b552fe9db readme wording 2 years ago
Brett Kuprel b7c2414c76 readme wording 2 years ago
Brett Kuprel 4e62f85ab9 readme wording 2 years ago
Brett Kuprel 53695e32f7 updated readme with torch examples 2 years ago
kuprel 9d5bb34df0 default to torch+mega in colab 2 years ago