Skip to content

Commit d55fbb1

Browse files
authored
fix transformers==4.33 (#1655)
1 parent b133bd2 commit d55fbb1

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

swift/llm/utils/argument.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1087,8 +1087,9 @@ def _init_training_args(self) -> None:
10871087
kwargs['neftune_noise_alpha'] = self.neftune_noise_alpha
10881088

10891089
parameters = inspect.signature(Seq2SeqTrainingArguments.__init__).parameters
1090-
if 'include_num_input_tokens_seen' in parameters:
1091-
kwargs['include_num_input_tokens_seen'] = self.include_num_input_tokens_seen
1090+
for k in ['lr_scheduler_kwargs', 'include_num_input_tokens_seen']:
1091+
if k in parameters:
1092+
kwargs[k] = getattr(self, k)
10921093
if 'eval_strategy' in parameters:
10931094
kwargs['eval_strategy'] = self.evaluation_strategy
10941095
else:
@@ -1106,7 +1107,6 @@ def _init_training_args(self) -> None:
11061107
num_train_epochs=self.num_train_epochs,
11071108
max_steps=self.max_steps,
11081109
lr_scheduler_type=self.lr_scheduler_type,
1109-
lr_scheduler_kwargs=self.lr_scheduler_kwargs,
11101110
warmup_ratio=self.warmup_ratio,
11111111
warmup_steps=self.warmup_steps,
11121112
logging_steps=self.logging_steps,

0 commit comments

Comments
 (0)