Script to log predictions grid to wandb.
830d7a2
Pedro Cuenca
commited on
Spaces:
Running
Running
Commit History
Merge pull request #32 from borisdayma/feat-model
699e1d9
fix: typo and missing tokenizer files
09362db
fix: missing arg
f65ccb3
fix: typo
a173dad
feat: restore state from checkpoint
4aced93
Merge branch 'add-tokenizer-save' into feat-model
28f08be
feat: model config not hardcoded
ad6ad64
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
Merge pull request #30 from pcuenca/clip-score
6567fd7
Tanishq Abraham
commited on
Merge pull request #29 from borisdayma/load_checkpoint
862924a
unverified
Score results using pre-trained CLIP.
5a9a1b6
Pedro Cuenca
commited on
fix: config used in preprocess
6d252e9
fix: custom reference
e803feb
fix: typo
ca83cca
feat: allow loading a model checkpoint
3d61350
feat: hardcode our full dataset
499ddb2
Merge pull request #25 from borisdayma/fix-config
d811136
unverified
feat: bigger warnings
62dad48
feat: use str mode for json
3e6ab1f
fix: import json
90320ea
fix: output directory must exist
6e89e9e
feat: update scriptst
63249ac
feat: update model config + save optim
a30dbd3
Merge pull request #28 from patil-suraj/tpu-demo
e31a84f
unverified
Ritobrata Ghosh
commited on
remove .ipynb
c879290
add tpu demo notebook
4b5a542
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
Merge pull request #26 from tmabraham/generation-training-demo
fc8c230
unverified
demo for generation, including during training from wandb artifact
c48da33
fix: model config
5aaf9df
Merge pull request #23 from khalidsaifullaah/main
eb591ff
unverified
pcuenca
commited on
