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

fix: chat_templates.py bug (#1048)

Browse source 
* fix: chat_template bug

* fix: check trainer attribute values are not None
This commit is contained in:
Nazim Ali 2024-09-22 04:18:37 -04:00 • committed by GitHub
commit 02cc97e388
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 3 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

6
unsloth/chat_templates.py
View file

@ -1842,10 +1842,10 @@ def train_on_responses_only(
return { "labels" : all_labels }
pass
if hasattr(trainer, "train_dataset"):
if hasattr(trainer, "train_dataset") and trainer.train_dataset is not None:
trainer.train_dataset = trainer.train_dataset.map(_train_on_responses_only, batched = True)
if hasattr(trainer, "eval_dataset"):
trainer.eval_dataset = trainer.eval_dataset .map(_train_on_responses_only, batched = True)
if hasattr(trainer, "eval_dataset") and trainer.eval_dataset is not None:
trainer.eval_dataset = trainer.eval_dataset.map(_train_on_responses_only, batched = True)
return trainer
pass