Commit History

fix: shampoo -> distributed shampoo
edae62d

boris commited on

feat: update params
604a65d

boris commited on

feat: add shampoo optimizer
0b87452

boris commited on

feat: update sweep
e1555d4

boris commited on

feat: create config files
dc5c024

boris commited on

feat: allow abstract_init
772415c

boris commited on

fix: typo
5c84978

boris commited on

feat: log more metrics
1b757dc

boris commited on

feat: shard by host is optional
901ff72

boris commited on

feat: load data first
fdf7698

boris commited on

feat: display local TPU's
15993e3

boris commited on

fix: check local TPU instances only
87fed1b

boris commited on

feat: add config
3f13951

boris commited on

feat(setup): require braceexpand for dev
98f1db7

boris commited on

fix(data): type
c6ebb14

boris commited on

feat(train): handle multi-hosts
5b533b5

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: update model name
61c93f2

boris commited on

fix(train): update model name
b257ca8

boris commited on

feat: log num_params
1f57ad7

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

feat(requirements): add pillow
26651dd

boris commited on

feat(sweep): update config
de21250

boris commited on

Merge pull request #114 from borisdayma/fix-110
caf7f44
unverified

boris commited on

feat: fix import error
3666d7b

boris commited on

Merge pull request #112 from borisdayma/’cleanup’
cbfa520
unverified

boris commited on

style: reformat
741bf32

boris commited on

feat: cleanup encode_dataset
ae754a3

boris commited on

feat: install black and isort
d2ec1ea

boris commited on

feat: add Makefile
6f19a1f

boris commited on

style: use isort
d209547

boris commited on

doc: setup instructions
fb1fbca

boris commited on

feat: add isort action
8b80a79

boris commited on

style: reformat per black
8db9ed4

boris commited on

feat: update black action
d54a7be

boris commited on

feat: add black action
5750492

boris commited on

feat(train): merge logged dict
baa52db

boris commited on

doc(README): update links
5f44d34

boris commited on

feat: add dev dependencies
2a3b8df

boris commited on

chore: move files
46f7469

boris commited on

chore: remove unused files
07eca3c

boris commited on