Commit Graph

298 Commits

Author SHA1 Message Date
Brett Kuprel 20e55522e2 Merge branch 'main' of https://github.com/kuprel/min-dalle 2022-07-04 22:05:41 -04:00
Brett Kuprel 990b7da701 update cogrun.py 2022-07-04 22:05:32 -04:00
Brett Kuprel 55fa76b798 explain super-condition factor 2022-07-04 21:42:27 -04:00
Brett Kuprel 8fe5a6f7ef update cogrun 2022-07-04 21:31:48 -04:00
Brett Kuprel 6d81d514a0 control super condition factor 2022-07-04 21:30:27 -04:00
Brett Kuprel f766a33dab super condition factor 2022-07-04 21:25:38 -04:00
Brett Kuprel 652341c02c remove image resize from colab 2022-07-04 20:56:45 -04:00
Brett Kuprel e1018d30f6 update replicate to match colab 2022-07-04 20:38:35 -04:00
Brett Kuprel b156e5c8c4 update replicate to match colab 2022-07-04 20:36:23 -04:00
Brett Kuprel 92b4dc3333 update cog 2022-07-04 20:10:37 -04:00
Brett Kuprel 18b2c84a57 update colab 2022-07-04 20:08:31 -04:00
Brett Kuprel a4de466817 update colab to use generate_image_stream 2022-07-04 20:06:28 -04:00
Brett Kuprel 5f4815775b generate_image_stream 2022-07-04 20:02:33 -04:00
Brett Kuprel cf5b116284 update predict.py 2022-07-04 18:41:00 -04:00
Brett Kuprel d3beb3ab57 update colab to match predict.py 2022-07-04 18:40:32 -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 41a9c9c69d update colab to properly handle intermediate_image_count=1 2022-07-04 17:44:30 -04:00
Brett Kuprel 25f5d8f800 update colab notebook to use log2_mid_count properly 2022-07-04 17:29:44 -04:00
Brett Kuprel 1702d3c439 log2_mid_count 2022-07-04 17:27:23 -04:00
Brett Kuprel 059d8f3448 update colab to use log2_mid_count 2022-07-04 17:27:02 -04:00
Brett Kuprel 8c068f0dcd display_size parameter in colab 2022-07-04 16:30:39 -04:00
Brett Kuprel 0d9998926d display intermediate images 2022-07-04 16:06:49 -04:00
Brett Kuprel b634375edf display intermediate images 2022-07-04 16:06:14 -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 e405577d99 4x4 works on colab T4 and P100 2022-07-04 10:21:12 -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 Kuprel 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
w4ffl35 7d549505da Fixes #16 - mega model running out of memory
Prior to this fix the mega model will (more often than not) fail when running in succession.

Clearing the cache seems to fix the issue.
2022-07-04 00:26:17 -06: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