SEDD small model for use with inference code in https://github.com/louaaron/Score-Entropy-Discrete-Diffusion