Use jax.device_count(), don't assume 8.
a11eff5
Pedro Cuenca
commited on
Spaces:
Running
Running
Commit History
Add missing import for CLIP.
7cc9b38
Pedro Cuenca
commited on
Score results using pre-trained CLIP.
5a9a1b6
Pedro Cuenca
commited on
remove .ipynb
c879290
add tpu demo notebook
4b5a542
fix forced bos token, also applying BART model to 8 samples now
8d4e13c
demo for generation, including during training from wandb artifact
c48da33
change bart-large-cnn to bart-large in demo folder
5801f13
Ritobrata Ghosh
commited on
Move generate nb by @ghosh-r to demo
dcbf091
Pedro Cuenca
commited on