Awesome

Code for reproduction

Model weights, sampling code and notebook to reproduce figures and tables

To rerun experiments with model on benchmark datasets run python -m run.run from this directory. To rerun RL Training run python -m run.run_rl.py from this directory.