dalle-mini / dalle_mini

Commit History

feat: allow abstract_init
772415c

boris commited on

feat: shard by host is optional
901ff72

boris commited on

fix(data): type
c6ebb14

boris commited on

feat: split shards by host
ed93c8a

boris commited on

style
a6252c9

boris commited on

feat(data): accept braceexpand notation
5ee6e60

boris commited on

feat: minor improvements
53dade7

boris commited on

fix(config): set min/max for generation
eb24dbc

boris commited on

fix: causal_mask based on image tokens
8654dc9

boris commited on

refactor(model): inherit from HF Flax & simplify
972bc8d

boris commited on

style
6f1f2d9

boris commited on

fix: adjust training script + dataloader
a96f4dc

boris commited on

fix(model): use correct params
a11892f

boris commited on

Merge branch 'main' of https://github.com/borisdayma/dalle-mini into add-custom-model
f234ccf

boris commited on

style: use isort
d209547

boris commited on

chore: remove unused files
07eca3c

boris commited on

feat(model): set default config for legacy models
92ccf4c

boris commited on

refactor: captioned_strip used only in gradio
0ca6514

boris commited on

feat: reorganize app
d4e833e

boris commited on

fix(data): minor bugs
0fe3e72

boris commited on

feat: handle data in separate file
85c1b8e

boris commited on

fix: correct use of dtype
b7d8724

boris commited on

feat: use model definition
803c7df

boris commited on

feat(text): more char
bf25d32

boris commited on

feat(text): improvements on pre-processing
7b58e88

boris commited on

feat(text): handle dates & prices
0465605

boris commited on

feat: keep %
229cdc0

boris commited on

feat(text): few improvements
849c5f3

boris commited on

feat(text): use hf_hub for wiki word count
a96c347

boris commited on

fix: typo
41b680b

boris commited on

feat: simplify fix_html
d054d1b

boris commited on

add property to get num params
e5a52b9

valhalla commited on

don't tie embeddings
a265819

valhalla commited on

make checkpointing optional
f6c4cb2

valhalla commited on

handle dtype for embeddings
29db327

valhalla commited on

add gradient checkpointing
95a8ed2

valhalla commited on

fix layernorm
7774483

valhalla commited on

add partition helpers
2856356

valhalla commited on

remove bias and minor fixes
180ed1e

valhalla commited on

add standalone modeling file
6197b2f

valhalla commited on

feat: add ftfy
a09ea25

boris commited on

feat: add text utilities
1212a74

boris commited on

Refactor: use VQGAN model from github, remove local copy
11ae595

Pedro Cuenca commited on

feat: allow display in multi lines
0dd7d80

boris commited on

refactor: move `captioned_strip` to library.
f62b045

Pedro Cuenca commited on

Get predictions from backend.
6e79248

Pedro Cuenca commited on

feat: separate model definition
7f962d6

boris commited on

chore: file not needed
231a81a

boris commited on

doc: reference vqgan-jax repo
cc3e85c

boris commited on

* dalle_mini package with models and utilities:
150ed18

Pedro Cuenca commited on