Skip to content

Commit 7e16e00

Browse files
committed
Merge branch 'main' into release/1.7
2 parents 3393361 + 8a128f2 commit 7e16e00

File tree

1 file changed

+8
-1
lines changed

1 file changed

+8
-1
lines changed

swift/trainers/mixin.py

Lines changed: 8 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -590,7 +590,14 @@ def create_optimizer(self):
590590
opt_model = self.model
591591

592592
if self.optimizer is None:
593-
decay_parameters = self.get_decay_parameter_names(opt_model)
593+
if version.parse(
594+
transformers.__version__) < version.parse('4.34.0'):
595+
logger.warning(
596+
f'If you are using lora+, please remember using transformers>=4.34.0, '
597+
f'but now is {transformers.__version__}')
598+
return super().create_optimizer()
599+
else:
600+
decay_parameters = self.get_decay_parameter_names(opt_model)
594601
if isinstance(self.model, SwiftModel):
595602
optimizer_grouped_parameters = self.model.create_optimizer_param_groups(
596603
lr=self.args.learning_rate,

0 commit comments

Comments
 (0)