Skip to content

Commit 7682d87

Browse files
committed
fix: code refactor
Signed-off-by: Mehant Kammakomati <[email protected]>
1 parent 2979b45 commit 7682d87

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tuning/sft_trainer.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -132,8 +132,8 @@ def train(
132132
odm_config = None
133133
if data_args.data_config_path:
134134
_dataconfig = load_and_validate_data_config(data_args.data_config_path)
135-
if _dataconfig["dataprocessor"]["type"] == "odm":
136-
odm_config = ODMConfig(odm=ODM(**_dataconfig["dataprocessor"]["odm"]))
135+
if _dataconfig.dataprocessor.type == "odm":
136+
odm_config = ODMConfig(odm=ODM(**_dataconfig.dataprocessor.odm.asdict()))
137137

138138
USE_ALORA = False
139139
try:

0 commit comments

Comments
 (0)