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

fix: address TrainerBuilderBase class variables to instance var

Browse Source
This commit is contained in:
NanoCode012
2025-05-22 18:46:03 +07:00
parent 79472241e8
commit 5346b99a88
1 changed files with 5 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
src/axolotl/core/trainer_builder/base.py
View File

@@ -48,17 +48,17 @@ with suppress(ImportError):
class TrainerBuilderBase(abc.ABC): class TrainerBuilderBase(abc.ABC):
"""Base class for trainer builder.""" """Base class for trainer builder."""
_train_dataset = None
_eval_dataset = None
_model_ref = None
_peft_config = None
def __init__(self, cfg, model, tokenizer, processor=None): def __init__(self, cfg, model, tokenizer, processor=None):
self.cfg = cfg self.cfg = cfg
self.model = model self.model = model
self.tokenizer = tokenizer self.tokenizer = tokenizer
self.processor = processor self.processor = processor
self._train_dataset = None
self._eval_dataset = None
self._model_ref = None
self._peft_config = None
# If the model supports tagging, add the axolotl tag. # If the model supports tagging, add the axolotl tag.
# This makes sure the tag is correctly pushed even if a user calls # This makes sure the tag is correctly pushed even if a user calls
# model.push_to_hub instead of trainer.push_to_hub. # model.push_to_hub instead of trainer.push_to_hub.