fix _get_per_token_logps_and_entropies return type#5456

kashif merged 3 commits intomainhuggingface/trl:mainfrom

kashif-patch-1huggingface/trl:kashif-patch-1Copy head branch name to clipboard