Commit History

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

feat: reorganize samples
0588e94

boris commited on

feat: add more samples
046ae75

boris commited on

feat: update wandb inference
dc79278

boris commited on

Merge pull request #91 from borisdayma/feat-inf
335110d
unverified

boris commited on

feat: more samples
2ef2966

boris commited on

Merge pull request #90 from borisdayma/feat-new
fdbe19f
unverified

boris commited on

feat: cleanup
91d8a29

boris commited on

feat: add sample
1d51d0b

boris commited on

fix: typo
c85fbb6

boris commited on

fix: pmap clip32
9a553a4

boris commited on

feat: cleanup
2d169e3

boris commited on

feat: allow latest version only
ff051c9

boris commited on

refactor: loop over runs
bf3640d

boris commited on

feat: create a table
38705a9

boris commited on

feat: add scoring
353365f

boris commited on

feat: remove cache before creating artifacts
5f6b691

boris commited on

feat: add functions
b8bbe68

boris commited on

feat: log to backend
378a628

boris commited on

fix: state.step type
47e006f

boris commited on

Merge branch 'main' of https://github.com/borisdayma/dalle-mini
5faf0fd

boris commited on

feat: get rid of global_step + log more metrics
4a4820f

boris commited on

fix(seq2seq): memory issue
708a42c

boris commited on

Merge pull request #88 from borisdayma/feat-cumul
272552a
unverified

boris commited on

add standalone modeling file
6197b2f

valhalla commited on

feat(gitignore): ignore compiled files and wandb
0b21fa5

boris commited on

feat: update scripts
5ca30e6

boris commited on

feat: use optax for gradient accumulation
69cf636

boris commited on

Merge pull request #71 from borisdayma/fix-opt_state
77657e6
unverified

Pedro Cuenca commited on

feat: no need for default values
a37cd75

boris commited on

feat: limit artifacts size
7253e56

boris commited on

feat: log epoch + check params
074c5e1

boris commited on

feat: update defaults
9ed6378

boris commited on

fix(seq2seq): normalize text
061c06b

boris commited on

fix(seq2seq): use streaming arg
0c992bd

boris commited on

fix: remove breakpoint
b75e0e9

boris commited on

feat: handle streaming
a96f44d

boris commited on

Merge branch 'main' of https://github.com/borisdayma/dalle-mini into fix-opt_state
39caefb

boris commited on

feat: add ftfy
a09ea25

boris commited on

Merge pull request #87 from borisdayma/feat-text
df2dbc7
unverified

boris commited on

feat: add text utilities
1212a74

boris commited on

feat(inference_notebook): dalle-mini is installable
1c83da9

boris commited on

Merge pull request #74 from borisdayma/feat-setup
27a3435
unverified

boris commited on

doc: update README
a8c579f

boris commited on

feat: requirements not needed
8d594fd

boris commited on