Spaces:

flax-community
/

dalle-mini

Running

App Files Files Community

Commit History

feat: sinkhorn in lse mode (#155)

00d4661
unverified

boris commited on Apr 4, 2022

feat(demo): update model

b9a1a7d

boris commited on Apr 2, 2022

fix: sinkformer gradient

eed4896

boris commited on Mar 31, 2022

feat(model): allow bias (#152)

361a994
unverified

boris commited on Mar 30, 2022

feat(train): google-cloud-storage is optional

02b2308

boris commited on Mar 29, 2022

feat(train): rename logged config

955dc20

boris commited on Mar 29, 2022

feat: add sinkformer + custom final ln + pre-ln (#151)

f139b0b
unverified

boris commited on Mar 29, 2022

feat: placeholders for more config

69bcbeb

boris commited on Mar 28, 2022

feat: add mini_glu config

a7e5050

boris commited on Mar 28, 2022

feat: force final ln in encoder

32f4ba5

boris commited on Mar 28, 2022

feat: allow more configurations

5bd4c20

boris commited on Mar 28, 2022

fix: DeepNet doesn't scale weights of embedding/output layers (#150)

503d6b4
unverified

Shuming Ma Shuming Ma commited on Mar 28, 2022

feat: remove unecessary LN

02824a7

boris commited on Mar 27, 2022

feat: update mini config

d9a16f2

boris commited on Mar 25, 2022

feat: add cogview

472c4cc

boris commited on Mar 25, 2022

fix(textnormalizer): consider utf8 on windows (#148)

3b8d8cb
unverified

illtellyoulater commited on Mar 25, 2022

feat: implement transformer variants (#144)

542378c
unverified

boris commited on Mar 24, 2022

feat(train): log norm and histograms (#143)

b7b619a
unverified

boris commited on Mar 19, 2022

feat(data): super conditioning (#141)

7939874
unverified

boris commited on Mar 17, 2022

feat: support pod (#139)

803ccbf
unverified

boris commited on Mar 14, 2022

fix: no gradient checkpointing for new model

2e02683

boris commited on Feb 23, 2022

feat: no gradient checkpointing for params init

b798ed3

boris commited on Feb 23, 2022

feat: update configs

79557f9

boris commited on Feb 23, 2022

feat(dev): require datasets

3d64598

boris commited on Feb 23, 2022

fix(train): consider schedule offset

bc4734f

boris commited on Feb 23, 2022

feat(train): local jax cache

9f5e879

boris commited on Feb 23, 2022

feat: add bucket reference to artifact

d368fb6

boris commited on Feb 23, 2022

style: lint

d5d442a

boris commited on Feb 22, 2022

feat: handle gradient checkpointing

5173ec7

boris commited on Feb 22, 2022

feat: load from bucket

1c4e839

boris commited on Feb 22, 2022

feat(train): save to bucket

50498e6

boris commited on Feb 22, 2022

feat: reduce artifact space + offset step

34cf91c

boris commited on Feb 22, 2022

feat(demo): update reference

e558000

boris commited on Feb 21, 2022

feat: restore weights on CPU

5f954fc

boris commited on Feb 21, 2022

feat(train): simplify tokenizer loading

4cb21dd

boris commited on Feb 21, 2022

doc: update README

db5a22a

boris commited on Feb 10, 2022

feat: cleanup notebook

5a390e8

boris commited on Feb 10, 2022

feat: wandb required for checkpoints

38c2c4e

boris commited on Feb 10, 2022

feat(demo): uncomment pip install

094e178

boris commited on Feb 10, 2022

feat: improve inference demo

35fe578

boris commited on Feb 10, 2022

fix: position embedding for generate method

ebac379

boris commited on Feb 10, 2022

feat(train): use compilation cache

da9367c

boris commited on Feb 7, 2022

fix: typo

68cc185

boris commited on Feb 7, 2022

fix: load from checkpoint

44b7c3e

boris commited on Feb 4, 2022

fix: style

d483294

boris commited on Feb 3, 2022

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

0a691de

boris commited on Feb 2, 2022

feat: log num_parameters early

7cfe576

boris commited on Feb 2, 2022

fix: distributed shampoo class

696422e

boris commited on Feb 2, 2022

feat: update distributed_shampoo

5996680

boris commited on Feb 2, 2022

feat(modeling): simplify abstract_init

fa72aa7

boris commited on Feb 2, 2022