Spaces:

flax-community
/

dalle-mini

Running

App Files Files Community

dalle-mini / tools

Commit History

Load tokenizer associated to the model checkpoint, if possible.

a77c0d4

Pedro Cuenca commited on Jan 16, 2022

Use model configuration unless a specific one is supplied.

5ec61cc

Pedro Cuenca commited on Jan 16, 2022

fix(inference): use float32 + flatten logits

71c4de3

boris commited on Jan 15, 2022

feat: update inference pipeline

af807f7

boris commited on Jan 14, 2022

style: isort

531cd78

boris commited on Jan 14, 2022

style: apply to distributed_shampoo

e669c1b

boris commited on Jan 14, 2022

feat: add best_effort_memory_usage_reduction

4d518c7

boris commited on Jan 14, 2022

feat: update distributed_shampoo

b90198c

boris commited on Jan 13, 2022

fix: weight decay Adam + speed logging

7143593

boris commited on Jan 12, 2022

feat: add micro config

e501f71

boris commited on Jan 10, 2022

fix: shampoo -> distributed shampoo

edae62d

boris commited on Jan 9, 2022

feat: update params

604a65d

boris commited on Jan 9, 2022

feat: add shampoo optimizer

0b87452

boris commited on Jan 8, 2022

feat: update sweep

e1555d4

boris commited on Jan 6, 2022

feat: create config files

dc5c024

boris commited on Jan 6, 2022

feat: allow abstract_init

772415c

boris commited on Jan 5, 2022

fix: typo

5c84978

boris commited on Jan 5, 2022

feat: log more metrics

1b757dc

boris commited on Jan 5, 2022

feat: shard by host is optional

901ff72

boris commited on Jan 3, 2022

feat: load data first

fdf7698

boris commited on Dec 24, 2021

feat: display local TPU's

15993e3

boris commited on Dec 24, 2021

fix: check local TPU instances only

87fed1b

boris commited on Dec 24, 2021

feat: add config

3f13951

boris commited on Dec 24, 2021

feat(train): handle multi-hosts

5b533b5

boris commited on Dec 24, 2021

style

a6252c9

boris commited on Dec 24, 2021

feat: minor improvements

53dade7

boris commited on Dec 21, 2021

fix: update model name

61c93f2

boris commited on Dec 20, 2021

fix(train): update model name

b257ca8

boris commited on Dec 20, 2021

feat: log num_params

1f57ad7

boris commited on Dec 20, 2021

fix: adjust training script + dataloader

a96f4dc

boris commited on Dec 19, 2021

feat(sweep): update config

de21250

boris commited on Nov 30, 2021

style: reformat

741bf32

boris commited on Nov 30, 2021

feat: cleanup encode_dataset

ae754a3

boris commited on Nov 30, 2021

style: use isort

d209547

boris commited on Nov 30, 2021

feat(train): merge logged dict

baa52db

boris commited on Nov 30, 2021

chore: move files

46f7469

boris commited on Nov 30, 2021

feat: add samples

e6c2573
unverified

boris commited on Nov 29, 2021

feat: add more samples

b3b48e3
unverified

boris commited on Nov 29, 2021

fix: correct clip params

5b16588

boris commited on Nov 28, 2021

feat(log_inference_samples): cleanup

cb127c4

boris commited on Nov 28, 2021

refactor: move to tools

4e4a30f

boris commited on Nov 28, 2021