Dia struggles with long pauses

[Blaizzy]

The model still struggles with excessively long pauses in some situations, especially the ellipsis break used in the example here: 'But the existential risks... ' -- I wonder if some post-processing of the logits to add a repetition penalty could help.

Originally posted by @lucasnewman in #99