Brett Kuprel
|
f0c4fc7350
|
fix individual images
|
2 years ago |
|
Brett Kuprel
|
798e6ac5a3
|
optionally tile images in token space
|
2 years ago |
|
Brett Kuprel
|
00d9363172
|
fixed bug on replicate
|
2 years ago |
|
Brett Kuprel
|
3c28b1059b
|
0.3.13, simplified code, specify device when initializing MinDalle
|
2 years ago |
|
Brett Kuprel
|
a2dca41623
|
update tkinter ui
|
2 years ago |
|
Brett Kuprel
|
d64e957731
|
add temperature parameter
|
2 years ago |
|
Brett Kuprel
|
2ab4365059
|
add performance graph to readme
|
2 years ago |
|
Brett Kuprel
|
1ffdef9a56
|
clamp in place
|
2 years ago |
|
Brett Kuprel
|
dba3f11b3f
|
faster inference with cuda/cudnn backends flags
|
2 years ago |
|
Brett Kuprel
|
703bfb231d
|
detach unnecessary
|
2 years ago |
|
Brett Kuprel
|
985c5fad4c
|
removing unused import
|
2 years ago |
|
Brett Kuprel
|
9eb5633931
|
clamp tokens to appropriate bounds
|
2 years ago |
|
Brett Kuprel
|
24162ee717
|
change grid_size argument of generate_images to image_count
|
2 years ago |
|
Brett Kuprel
|
736904ef2f
|
fix typing
|
2 years ago |
|
Brett Kuprel
|
2cac9220b5
|
generate_images_stream and generate_images
|
2 years ago |
|
Brett Kuprel
|
c199507a7a
|
vqgan needs to be float32
|
2 years ago |
|
Brett Kuprel
|
da62298f06
|
support bfloat16
|
2 years ago |
|
Brett Kuprel
|
f071b31bdd
|
properly limit input to 64 tokens
|
2 years ago |
|
Brett Kuprel
|
89a125b4b9
|
control top_k value
|
2 years ago |
|
Brett Kuprel
|
ba6080668f
|
update readme
|
2 years ago |
|
Brett Kuprel
|
9a4d9dc118
|
remove unused imports
|
2 years ago |
|
Brett Kuprel
|
ccdcbc7d46
|
forgot missing 2**
|
2 years ago |
|
Brett Kuprel
|
6d81d514a0
|
control super condition factor
|
2 years ago |
|
Brett Kuprel
|
5f4815775b
|
generate_image_stream
|
2 years ago |
|
Brett Kuprel
|
97d22f0fdd
|
handle log2_mid_count=0
|
2 years ago |
|
Brett Kuprel
|
1702d3c439
|
log2_mid_count
|
2 years ago |
|
Brett Kuprel
|
0d9998926d
|
display intermediate images
|
2 years ago |
|
Brett Kuprel
|
deefd24919
|
decode_row
|
2 years ago |
|
Brett Kuprel
|
6f617fe98f
|
inplace attention state, faster and less memory
|
2 years ago |
|
Brett Kuprel
|
377d15cb16
|
faster decoder self attention
|
2 years ago |
|
Brett Kuprel
|
256fd1aab0
|
empty cache if cuda available
|
2 years ago |
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.
|
2 years ago |
|
Brett Kuprel
|
313635e914
|
fixed typing error for older python versions
|
2 years ago |
|
Brett Kuprel
|
1eb56737d8
|
added grid_size parameter to generate a grid of images
|
2 years ago |
|
Brett Kuprel
|
cf9656baa2
|
added is_verbose flag
|
2 years ago |
|
Brett Kuprel
|
35e97768a5
|
v0.2.0, MinDalleTorch -> MinDalle, breaking change
|
2 years ago |
|
Brett Kuprel
|
fff44d683e
|
update readme, random seed if none specified
|
2 years ago |
|
Brett Kuprel
|
be2beca7c0
|
added to pypi
|
2 years ago |
|
Brett Kuprel
|
69683fec94
|
simplify import
|
2 years ago |
|
Brett Kuprel
|
18c72ed34d
|
simplified MinDalleTorch
|
2 years ago |
|
Brett Kuprel
|
07ce93d5f8
|
moved flax model and conversion code to separate repository
|
2 years ago |
|
Brett Kuprel
|
85f5866eff
|
remove config.json dependency, default to torch in image_from_text.py
|
2 years ago |
|
Brett Kuprel
|
09a0f85b8e
|
separate setup processes for flax and torch
|
2 years ago |
|
Brett Kuprel
|
7bf76deafb
|
fixed wrong file path
|
2 years ago |
|
Brett Kuprel
|
e4c2be54cb
|
save converted detokenizer params
|
2 years ago |
|
Brett Kuprel
|
b40fd83a0d
|
mega works with latest flax version 0.5.2 now, removing 0.4.2 pin
|
2 years ago |
|
Brett Kuprel
|
08b158d580
|
updated readme
|
2 years ago |
|
Brett Kuprel
|
2311a1af7b
|
delete cache
|
2 years ago |
|
Brett Kuprel
|
b913b58353
|
pre converting params to torch allows mega to run in standard colab runtime
|
2 years ago |
|
Brett Kuprel
|
c2a3858c96
|
delete params sooner
|
2 years ago |
