Learning to vary: Teaching LMs to reproduce human linguistic variability in next-word prediction