Generation utils for pytorch transformers

I am looking for something similar. Were you able to find a solution for this?