Brett Kuprel
|
377d15cb16
|
faster decoder self attention
|
2022-07-04 08:05:55 -04:00 |
|
Brett Kuprel
|
256fd1aab0
|
empty cache if cuda available
|
2022-07-04 07:21:54 -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
|
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
|
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
|
1e18ba0ffa
|
is_expendable argument reduces memory usage for command line script
|
2022-06-30 06:43:10 -04:00 |
|
Brett Kuprel
|
61cc99c13c
|
read tokenizer files with utf8 encoding
|
2022-06-29 14:18:23 -04:00 |
|
Brett Kuprel
|
ed91ab4a30
|
refactored to load models once and run multiple times
|
2022-06-29 09:42:12 -04:00 |
|