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
|
1ffdef9a56
|
clamp in place
|
2 years ago |
|
Brett Kuprel
|
dba3f11b3f
|
faster inference with cuda/cudnn backends flags
|
2 years ago |
|
Brett Kuprel
|
9eb5633931
|
clamp tokens to appropriate bounds
|
2 years ago |
|
Brett Kuprel
|
736904ef2f
|
fix typing
|
2 years ago |
|
Brett Kuprel
|
da62298f06
|
support bfloat16
|
2 years ago |
|
Brett Kuprel
|
89a125b4b9
|
control top_k value
|
2 years ago |
|
Brett Kuprel
|
ccdcbc7d46
|
forgot missing 2**
|
2 years ago |
|
Brett Kuprel
|
6d81d514a0
|
control super condition factor
|
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
|
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
|
35e97768a5
|
v0.2.0, MinDalleTorch -> MinDalle, breaking change
|
2 years ago |
|
Brett Kuprel
|
be2beca7c0
|
added to pypi
|
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
|
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
|
2311a1af7b
|
delete cache
|
2 years ago |
|
Brett Kuprel
|
df9aa6f915
|
sort -> topk, prev_token_and_index -> prev_token, token_index
|
2 years ago |
|
Brett Kuprel
|
fb97ba5e20
|
update readme, cleanup
|
2 years ago |
|
Brett Kuprel
|
d99828a239
|
simplified flax attention and matched torch attention
|
2 years ago |
|
Brett Kuprel
|
661ec976ac
|
simplified attention for torch model
|
2 years ago |
|
Brett Kuprel
|
ed91ab4a30
|
refactored to load models once and run multiple times
|
2 years ago |
|
Brett Kuprel
|
17c96fe110
|
works with cuda
|
2 years ago |
|
Brett Kuprel
|
9d6b6dcc92
|
previous commit broke flax model, fixed now
|
2 years ago |
|
Brett Kuprel
|
5aa6fe49bf
|
use cuda if available
|
2 years ago |
|
Brett Kuprel
|
aef24ea157
|
torch.no_grad(), cleanup
|
2 years ago |
|
Brett Kuprel
|
34df2b97df
|
previous commit broke colab example, so adjusting flax requirement to 0.4.2 for now
|
2 years ago |
|
Brett Kuprel
|
38ebe54a38
|
works with latest flax version 0.5.2 now
|
2 years ago |
|
Brett Kuprel
|
e7001f063c
|
simplified
|
2 years ago |
|
Brett Kuprel
|
18e6a9852f
|
license and cleanup
|
2 years ago |
|
Brett Kuprel
|
c936d26102
|
back to linear attention
|
2 years ago |
|
Brett Kuprel
|
018414a5c3
|
fixed relative imports
|
2 years ago |
