Official implementation of Selective Token Generation (COLING'22) - View it on GitHub