Add warmup to config

[Randl]

Somehow https://github.com/huggingface/alignment-handbook/pull/31 didn't add it in the end