Merge pull request #37 from tmabraham/add-gradio-demo
7e6e1fe
Tanishq Abraham
commited on
Spaces:
Running
Running
Commit History
add gradio demo
6b0d541
Merge pull request #36 from borisdayma/chore-cleanup
21a35bc
chore: move requirements to correct section
469b2c2
chore: file not needed
231a81a
doc: reference vqgan-jax repo
cc3e85c
doc: update README
5542365
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
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