fix: peft bug fix inside sequence classification recipe (#781)

adil-a · web-flow · commit a198c9dc82e7 · 2025-11-07T15:25:57.000-05:00
Signed-off-by: adil-a &lt;adil.asif2000@hotmail.com&gt;
diff --git a/nemo_automodel/recipes/llm/train_seq_cls.py b/nemo_automodel/recipes/llm/train_seq_cls.py
@@ -81,7 +81,7 @@ def setup(self):
             self.cfg.get("checkpoint", None),
             self.cfg.get("model.cache_dir", None),
             self.cfg.model.pretrained_model_name_or_path,
-            False,
+            True if self.cfg.get("peft", None) else False,
         )
 
         from nemo_automodel.components.checkpoint.checkpointing import Checkpointer

Original file line number	Diff line number	Diff line change
`@@ -81,7 +81,7 @@ def setup(self):`
`81`	`81`	`self.cfg.get("checkpoint", None),`
`82`	`82`	`self.cfg.get("model.cache_dir", None),`
`83`	`83`	`self.cfg.model.pretrained_model_name_or_path,`
`84`		`- False,`
	`84`	`+ True if self.cfg.get("peft", None) else False,`
`85`	`85`	`)`
`86`	`86`
`87`	`87`	`from nemo_automodel.components.checkpoint.checkpointing import Checkpointer`