ceos/unsloth
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Update tokenizer_utils.py

Browse source
This commit is contained in:
Daniel Han 2024-11-09 17:40:32 -08:00
commit 61665e96d8
1 changed files with 1 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

4
unsloth/tokenizer_utils.py
View file

@ -1001,7 +1001,7 @@ def patch_sft_trainer_tokenizer():
# Also DPO weirdly tokenizes non numeric columns? Delete them!
check_text += \
"\n"\
"if hasattr(self.train_dataset, 'column_names'):\n"
"if hasattr(self.train_dataset, 'column_names'):\n"\
" column_names = set(self.train_dataset.column_names)\n"\
" check = ['chosen', 'rejected', 'prompt', 'chosen_input_ids', 'chosen_attention_mask',\n"\
" 'chosen_labels', 'rejected_input_ids', 'rejected_attention_mask', 'rejected_labels',\n"\
@ -1015,8 +1015,6 @@ def patch_sft_trainer_tokenizer():
check_text = "\n".join(" "*where + x for x in check_text)
function = function.replace(replacer, check_text + replacer)
print(function)
raise
exec(function, globals())
exec(f"trl.trainer.{path_to_trainer}.{function_name} = {function_name}", globals())