Commit Graph
100 Commits
Author SHA1 Message Date
Brett Kuprel cf5b116284 update predict.py 2022-07-04 18:41:00 -04:00
Brett Kuprel 0ac2b07344 update cog.yaml 2022-07-04 18:39:16 -04:00
Brett Kuprel 73f4a47fb5 update predict.py 2022-07-04 18:37:07 -04:00
Brett Kuprel 970afc9631 Merge branch 'main' of https://github.com/kuprel/min-dalle 2022-07-04 17:46:38 -04:00
Brett Kuprel 97d22f0fdd handle log2_mid_count=0 2022-07-04 17:46:31 -04:00
Brett Kuprel 1702d3c439 log2_mid_count 2022-07-04 17:27:23 -04:00
Brett Kuprel 0d9998926d display intermediate images 2022-07-04 16:06:49 -04:00
Brett Kuprel 672f325b3b 14s now for 4x4 on A100 2022-07-04 11:19:39 -04:00
Brett Kuprel 8f7f0ba756 update readme 2022-07-04 10:54:25 -04:00
Brett Kuprel ecc1534f87 5x5 grid works on colab P100 so should work on replicate 2022-07-04 10:28:47 -04:00
Brett Kuprel 3dc92b19bd update readme 2022-07-04 10:22:45 -04:00
Brett Kuprel 486c73c741 update cog.yaml 2022-07-04 10:06:09 -04:00
Brett Kuprel deefd24919 decode_row 2022-07-04 09:58:00 -04:00
Brett Kuprel 884202239f update notebook 2022-07-04 09:29:54 -04:00
Brett Kuprel e0fd4d4bff update cog.yaml 2022-07-04 09:21:31 -04:00
Brett Kuprel 6f617fe98f inplace attention state, faster and less memory 2022-07-04 09:14:37 -04:00
Brett Kuprel aca617dc64 new decoder self attention reduced memory usage and 3x3 grids now work on T4 in colab 2022-07-04 08:25:50 -04:00
Brett Kuprel 461890486e update python version to 0.2.15 2022-07-04 08:10:27 -04:00
Brett Kuprel 377d15cb16 faster decoder self attention 2022-07-04 08:05:55 -04:00
Brett Kuprel a79d30f718 png -> jpg 2022-07-04 07:38:16 -04:00
Brett Kuprel a21ea6ea14 update replicate, clear cuda cache if cuda available 2022-07-04 07:28:44 -04:00
Brett Kuprel 256fd1aab0 empty cache if cuda available 2022-07-04 07:21:54 -04:00
Brett KuprelandGitHub 9602055eaa Merge pull request #57 from w4ffl35/bugfix-for-mega-model
Fixes #16 - mega model running out of memory
2022-07-04 07:16:39 -04:00
Brett Kuprel 6b433d5ba6 update colab 2022-07-03 18:40:27 -04:00
Brett Kuprel 53741154c7 update readme 2022-07-03 17:29:53 -04:00
Brett Kuprel aa08f7f93c update readme 2022-07-03 16:49:38 -04:00
Brett Kuprel ec14eeba62 convert pngs to jpgs 2022-07-03 16:30:36 -04:00
Brett Kuprel 2661074589 update readme 2022-07-03 16:26:49 -04:00
Brett Kuprel 6ce60179ff update cog.yaml 2022-07-03 16:16:23 -04:00
Brett Kuprel 948e05c0dd update readme 2022-07-03 15:40:58 -04:00
Brett Kuprel a5004efad0 update readme 2022-07-03 13:35:36 -04:00
Brett Kuprel b939a1e5ac update cog.yaml 2022-07-03 12:20:03 -04:00
Brett Kuprel f0f9c5e83e update replicate predict.py 2022-07-03 12:14:24 -04:00
Brett Kuprel 8d808599d3 h100 stat tbd 2022-07-03 11:53:45 -04:00
Brett Kuprel 229af7233d h100 stat tbd 2022-07-03 11:34:58 -04:00
Brett Kuprel 107213b28b 3x3 only works on colab if P100 is allocated 2022-07-03 11:28:28 -04:00
Brett Kuprel f1bb562ac1 update readme 2022-07-03 11:22:52 -04:00
Brett Kuprel 134fdaf76b add nvidia-smi to notebook 2022-07-03 11:17:37 -04:00
Brett Kuprel 9bdba57195 update replicate 2022-07-02 16:36:51 -04:00
Brett Kuprel 9e146e6675 update readme 2022-07-02 14:56:42 -04:00
Brett Kuprel cbee448f11 update notebook 2022-07-02 14:54:18 -04:00
Brett Kuprel a22d510270 update readme 2022-07-02 13:06:48 -04:00
Brett Kuprel d8a0cbcaf5 update readme 2022-07-02 13:06:03 -04:00
Brett Kuprel 1351478ee5 update readme 2022-07-02 12:55:59 -04:00
Brett Kuprel 05d9d5f07f update readme 2022-07-02 12:34:11 -04:00
Brett Kuprel bf1acab78a update readme 2022-07-02 12:32:54 -04:00
Brett Kuprel ec37a2dafe update replicate cog yaml file 2022-07-02 12:02:55 -04:00
Brett Kuprel e8dac15810 update replicate max grid size 2022-07-02 11:59:26 -04:00
Brett Kuprel 31fd7c231d updated replicate cog.yaml python packages 2022-07-02 11:23:32 -04:00
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