We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 3f09f09 commit bbd5436Copy full SHA for bbd5436
1 file changed
tuning/sft_trainer.py
@@ -151,8 +151,10 @@ def train(
151
odm_config = None
152
if data_args.data_config_path:
153
_dataconfig = load_and_validate_data_config(data_args.data_config_path)
154
- _dataconfig.dataprocessor.odm["resume_from_checkpoint"] = resume_from_checkpoint
155
if _dataconfig.dataprocessor.type == "odm":
+ _dataconfig.dataprocessor.odm[
156
+ "resume_from_checkpoint"
157
+ ] = resume_from_checkpoint
158
odm_config = ODMConfig(odm=ODM(**_dataconfig.dataprocessor.odm))
159
160
USE_ALORA = False
0 commit comments