Skip to content

Commit bbd5436

Browse files
committed
feat: resume functionality
Signed-off-by: Mehant Kammakomati <mehant.kammakomati2@ibm.com>
1 parent 3f09f09 commit bbd5436

1 file changed

Lines changed: 3 additions & 1 deletion

File tree

tuning/sft_trainer.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -151,8 +151,10 @@ def train(
151151
odm_config = None
152152
if data_args.data_config_path:
153153
_dataconfig = load_and_validate_data_config(data_args.data_config_path)
154-
_dataconfig.dataprocessor.odm["resume_from_checkpoint"] = resume_from_checkpoint
155154
if _dataconfig.dataprocessor.type == "odm":
155+
_dataconfig.dataprocessor.odm[
156+
"resume_from_checkpoint"
157+
] = resume_from_checkpoint
156158
odm_config = ODMConfig(odm=ODM(**_dataconfig.dataprocessor.odm))
157159

158160
USE_ALORA = False

0 commit comments

Comments
 (0)