Remove use_peft parameter from training script

Instead we make the PEFT config optional which has the same effect.
This commit is contained in:
nemo
2025-10-14 15:35:38 +02:00
parent 02408fb244
commit 20d4b5d347
2 changed files with 2 additions and 3 deletions
+1 -2
View File
@@ -63,8 +63,7 @@ class TrainPipelineConfig(HubMixin):
scheduler: LRSchedulerConfig | None = None
eval: EvalConfig = field(default_factory=EvalConfig)
wandb: WandBConfig = field(default_factory=WandBConfig)
use_peft: bool = False
peft: PeftConfig = field(default_factory=PeftConfig)
peft: PeftConfig | None = None
def __post_init__(self):
self.checkpoint_path = None
+1 -1
View File
@@ -247,7 +247,7 @@ def train(cfg: TrainPipelineConfig):
ds_meta=dataset.meta,
)
if cfg.use_peft:
if cfg.peft is not None:
logging.info("Using PEFT! Wrapping model.")
policy = wrap_policy_in_peft_model(cfg, policy)