Add sampling

[kingoflolz]

Training model rn to see if its even slightly sensible

[sdtblck]

let's get it actually working before we merge into main haha

[leogao2]

Requirements is missing imageio, tensorflow_datasets, transformers, cloud-tpu-client

[sdtblck]

This should be ready to merge now, but i'm just going to train some small models to make sure everything's working.

[CLAassistant]

Thank you for your submission! We really appreciate it. Like many open source projects, we ask that you all sign our Contributor License Agreement before we can accept your contribution.
0 out of 6 committers have signed the CLA.

❌ sid
❌ kingoflolz
❌ ConnorJL
❌ lucidrains
❌ leogao2
❌ sdtblck

---

sid seems not to be a GitHub user. You need a GitHub account to be able to sign the CLA. If you have already a GitHub account, please add the email address used for this commit to your account.
_{You have signed the CLA already but the status is still pending? Let us recheck it.}