tocmo0nlord/axolotl
1
0
Fork 0
Code Issues Pull Requests Actions 3 Packages Projects Releases Wiki Activity

chore: lint

Browse Source
This commit is contained in:
NanoCode012
2025-05-22 17:44:30 +07:00
parent b87850e11b
commit 24b61c1b67
2 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/axolotl/core/trainer_builder/rl.py
View File

@@ -197,7 +197,7 @@ class HFRLTrainerBuilder(TrainerBuilderBase):
trainer_cls, trainer_cls_args = self._get_trainer_cls(trainer_kwargs) trainer_cls, trainer_cls_args = self._get_trainer_cls(trainer_kwargs)
sig = inspect.signature(trainer_cls) sig = inspect.signature(trainer_cls)
if "tokenizer" in sig.parameters.keys(): if "tokenizer" in sig.parameters:
trainer_kwargs["tokenizer"] = self.tokenizer trainer_kwargs["tokenizer"] = self.tokenizer
else: else:
trainer_kwargs["processing_class"] = self.tokenizer trainer_kwargs["processing_class"] = self.tokenizer

2
src/axolotl/core/trainer_builder/sft.py
View File

@@ -400,7 +400,7 @@ class HFCausalTrainerBuilder(TrainerBuilderBase):
**data_collator_kwargs, **data_collator_kwargs,
) )
sig = inspect.signature(trainer_cls) sig = inspect.signature(trainer_cls)
if "processing_class" in sig.parameters.keys(): if "processing_class" in sig.parameters:
trainer_kwargs["processing_class"] = self.tokenizer trainer_kwargs["processing_class"] = self.tokenizer
else: else:
trainer_kwargs["tokenizer"] = self.tokenizer trainer_kwargs["tokenizer"] = self.tokenizer