Default parameters

[daniel-z-kaplan]

I am trying to run this repo with KL mode, in order to replicate both your results and SD results.
However, running the code on TPUv4 is going instantly OOM. Therefore, I'd like to request that you update the repository with the exact settings used for the results listed (3.3ish FID for KL most important to me), to make sure I can also replicate.