Commit History

chore: file not needed
231a81a

boris commited on

doc: reference vqgan-jax repo
cc3e85c

boris commited on

doc: update README
5542365

boris commited on

Add a couple of sliders and prevent generating without a prompt
0e8338d

Pedro Cuenca commited on

Update requirements
7158e2e

Pedro Cuenca commited on

Add dalle_mini directory module.
adfe05e

Pedro Cuenca commited on

Barebones demo app for local testing.
cb2ac60

Pedro Cuenca commited on

Script to log predictions grid to wandb.
830d7a2

Pedro Cuenca commited on

Merge pull request #32 from borisdayma/feat-model
699e1d9

boris commited on

fix: typo and missing tokenizer files
09362db

boris commited on

fix: missing arg
f65ccb3

boris commited on

fix: typo
a173dad

boris commited on

feat: restore state from checkpoint
4aced93

boris commited on

Merge branch 'add-tokenizer-save' into feat-model
28f08be

boris commited on

feat: model config not hardcoded
ad6ad64

boris commited on

Use jax.device_count(), don't assume 8.
a11eff5

Pedro Cuenca commited on

Add missing import for CLIP.
7cc9b38

Pedro Cuenca commited on

add tokenizer save to wandb:
aecf3a7

tmabraham commited on

Merge pull request #30 from pcuenca/clip-score
6567fd7

Tanishq Abraham commited on

Merge pull request #29 from borisdayma/load_checkpoint
862924a
unverified

boris commited on

Score results using pre-trained CLIP.
5a9a1b6

Pedro Cuenca commited on

fix: config used in preprocess
6d252e9

boris commited on

fix: custom reference
e803feb

boris commited on

fix: typo
ca83cca

boris commited on

feat: allow loading a model checkpoint
3d61350

boris commited on

feat: hardcode our full dataset
499ddb2

boris commited on

Merge pull request #25 from borisdayma/fix-config
d811136
unverified

boris commited on

feat: bigger warnings
62dad48

boris commited on

feat: use str mode for json
3e6ab1f

boris commited on

fix: import json
90320ea

boris commited on

fix: output directory must exist
6e89e9e

boris commited on

feat: update scriptst
63249ac

boris commited on

feat: update model config + save optim
a30dbd3

boris commited on

Merge pull request #28 from patil-suraj/tpu-demo
e31a84f
unverified

Ritobrata Ghosh commited on

remove .ipynb
c879290

valhalla commited on

add tpu demo notebook
4b5a542

valhalla commited on

Merge pull request #27 from tmabraham/fix-forced-bos-token-on-demo
8f484d9
unverified

pcuenca commited on

fix forced bos token, also applying BART model to 8 samples now
8d4e13c

tmabraham commited on

Merge pull request #26 from tmabraham/generation-training-demo
fc8c230
unverified

boris commited on

demo for generation, including during training from wandb artifact
c48da33

tmabraham commited on

fix: model config
5aaf9df

boris commited on

Merge pull request #23 from khalidsaifullaah/main
eb591ff
unverified

pcuenca commited on

Merge pull request #24 from borisdayma/feat--log-model-frequently
648e404
unverified

boris commited on

feat: use bart-large-cnn
19d68bb

boris commited on

fix: log metadata
99a1ff5

boris commited on

fix: define function before it is used
d449092

boris commited on

fix: correct arg
283adc6

boris commited on

feat: save model frequently
754f876

boris commited on

feat: split script for small and big runs
5e244d0

boris commited on

feat: update test script
3cccb01

boris commited on