Spaces:

flax-community
/

dalle-mini

Running

App Files Files Community

dalle-mini / demo /tpu-demo.ipynb

Pedro Cuenca

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

a11eff5 over 3 years ago

4.44 MB

Rendering notebook...