Brett Kuprel
|
f0c4fc7350
|
fix individual images
|
2022-07-17 08:34:32 -04:00 |
|
Brett Kuprel
|
798e6ac5a3
|
optionally tile images in token space
|
2022-07-17 07:33:31 -04:00 |
|
Brett Kuprel
|
00d9363172
|
fixed bug on replicate
|
2022-07-15 21:21:16 -04:00 |
|
Brett Kuprel
|
3c28b1059b
|
0.3.13, simplified code, specify device when initializing MinDalle
|
2022-07-15 17:18:23 -04:00 |
|
Brett Kuprel
|
a2dca41623
|
update tkinter ui
|
2022-07-15 09:00:26 -04:00 |
|
Brett Kuprel
|
d64e957731
|
add temperature parameter
|
2022-07-11 12:28:20 -04:00 |
|
Brett Kuprel
|
2ab4365059
|
add performance graph to readme
|
2022-07-11 08:39:48 -04:00 |
|
Brett Kuprel
|
1ffdef9a56
|
clamp in place
|
2022-07-10 08:07:54 -04:00 |
|
Brett Kuprel
|
dba3f11b3f
|
faster inference with cuda/cudnn backends flags
|
2022-07-09 06:48:51 -04:00 |
|
Brett Kuprel
|
703bfb231d
|
detach unnecessary
|
2022-07-08 13:26:53 -04:00 |
|
Brett Kuprel
|
985c5fad4c
|
removing unused import
|
2022-07-08 09:50:13 -04:00 |
|
Brett Kuprel
|
9eb5633931
|
clamp tokens to appropriate bounds
|
2022-07-08 09:19:28 -04:00 |
|
Brett Kuprel
|
24162ee717
|
change grid_size argument of generate_images to image_count
|
2022-07-07 20:21:43 -04:00 |
|
Brett Kuprel
|
736904ef2f
|
fix typing
|
2022-07-07 17:18:30 -04:00 |
|
Brett Kuprel
|
2cac9220b5
|
generate_images_stream and generate_images
|
2022-07-07 17:03:47 -04:00 |
|
Brett Kuprel
|
c199507a7a
|
vqgan needs to be float32
|
2022-07-07 08:53:27 -04:00 |
|
Brett Kuprel
|
da62298f06
|
support bfloat16
|
2022-07-07 08:21:20 -04:00 |
|
Brett Kuprel
|
f071b31bdd
|
properly limit input to 64 tokens
|
2022-07-05 22:14:19 -04:00 |
|
Brett Kuprel
|
89a125b4b9
|
control top_k value
|
2022-07-05 17:23:05 -04:00 |
|
Brett Kuprel
|
ba6080668f
|
update readme
|
2022-07-05 07:44:42 -04:00 |
|
Brett Kuprel
|
9a4d9dc118
|
remove unused imports
|
2022-07-05 06:31:01 -04:00 |
|
Brett Kuprel
|
ccdcbc7d46
|
forgot missing 2**
|
2022-07-04 23:29:48 -04:00 |
|
Brett Kuprel
|
6d81d514a0
|
control super condition factor
|
2022-07-04 21:30:27 -04:00 |
|
Brett Kuprel
|
5f4815775b
|
generate_image_stream
|
2022-07-04 20:02:33 -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
|
deefd24919
|
decode_row
|
2022-07-04 09:58:00 -04:00 |
|
Brett Kuprel
|
6f617fe98f
|
inplace attention state, faster and less memory
|
2022-07-04 09:14:37 -04:00 |
|
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
|
fff44d683e
|
update readme, random seed if none specified
|
2022-07-01 18:50:11 -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 Kuprel
|
07ce93d5f8
|
moved flax model and conversion code to separate repository
|
2022-07-01 14:06:50 -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
|
09a0f85b8e
|
separate setup processes for flax and torch
|
2022-07-01 11:08:33 -04:00 |
|
Brett Kuprel
|
7bf76deafb
|
fixed wrong file path
|
2022-07-01 10:58:29 -04:00 |
|
Brett Kuprel
|
e4c2be54cb
|
save converted detokenizer params
|
2022-07-01 10:17:29 -04:00 |
|
Brett Kuprel
|
b40fd83a0d
|
mega works with latest flax version 0.5.2 now, removing 0.4.2 pin
|
2022-07-01 02:58:43 -04:00 |
|
Brett Kuprel
|
08b158d580
|
updated readme
|
2022-06-30 16:50:04 -04:00 |
|
Brett Kuprel
|
2311a1af7b
|
delete cache
|
2022-06-30 15:48:20 -04:00 |
|
Brett Kuprel
|
b913b58353
|
pre converting params to torch allows mega to run in standard colab runtime
|
2022-06-30 14:54:08 -04:00 |
|
Brett Kuprel
|
c2a3858c96
|
delete params sooner
|
2022-06-30 11:44:36 -04:00 |
|