Commit Graph

240 Commits

Author SHA1 Message Date
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
Brett Kuprel 948e05c0dd update readme 2022-07-03 15:40:58 -04:00
Brett Kuprel e20c4a8c9f updated colab 2022-07-03 15:35:33 -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