Update rl_replacements.py

This commit is contained in:
Daniel Han 2025-02-11 21:14:41 -08:00
commit bbb6d1589d

View file

@ -78,7 +78,7 @@ def sft_trainer_prepare_dataset(function_name, function):
" add_special_tokens = False if has_bos_token_already else add_special_tokens\n"
check_text = check_text.split("\n")
check_text = "\n".join(" "*where + x for x in check_text)
check_text = "\n".join(" "*4 + x for x in check_text)
check_text = check_text.rstrip() + "\n"
# .*? matches first match. .+? matches final match.