Hello, Thank you for this implementation, it is nice being able to experiment with things, even without GPUs at hand. Would you mind implementing the repetition penalty? It seems to produce better/more consistent results...