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