dalle-mini / demo /tpu-demo.ipynb
Pedro Cuenca
Use jax.device_count(), don't assume 8.
a11eff5
raw
history
4.44 MB
Open in Colab