dalle-mini / dev

Commit History

fix: fixes training script
0df810d

boris commited on

fix: log train_metric only if defined
9bf9397

boris commited on

feat: add metrics + cleanup
6523a6d

boris commited on

feat: simplify parameters
87fac28

boris commited on

feat: use model definition
803c7df

boris commited on

fix(inference): update flax version (temporary fix)
6aa30f5

boris commited on

Merge pull request #93 from borisdayma/inference
e226ca6
unverified

boris commited on

feat: update samples
39d3a15

boris commited on

feat: update samples
8444c1b

boris commited on

feat: rename variables
c7fe380

boris commited on

feat: reorganize samples
b1aaa0f

boris commited on

fix: duplicate samples
7b8c2cb

boris commited on

fix: OOM
86c6c90

boris commited on

fix: OOM with checkpoints
e2400cc

boris commited on

fix: comment
36cb737

boris commited on

feat: cleanup training script
3cd6d41

boris commited on

feat: more samples
e3f152e

boris commited on

feat: add customization parameters
23b8870

boris 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

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

feat: update scripts
5ca30e6

boris commited on

feat: use optax for gradient accumulation
69cf636

boris 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