Text generation using language models with multiple exit heads - View it on GitHub