Move _get_kl_completion_ids into _get_kl_dataset#6181

qgallouedec wants to merge 3 commits into

mainhuggingface/trl:mainfrom

move-get-kl-datasethuggingface/trl:move-get-kl-datasetCopy head branch name to clipboard