Skip to content

Commit 7cf8206

Browse files
authored
✏️ fix typo (#1064)
1 parent e349174 commit 7cf8206

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

swift/llm/sft.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -249,12 +249,12 @@ def llm_sft(args: SftArguments) -> Dict[str, Union[str, Any]]:
249249
padding_to = args.max_length if args.sft_type == 'longlora' else None
250250
data_collator = partial(template.data_collator, padding_to=padding_to)
251251

252-
trian_batch_size = args.batch_size
252+
train_batch_size = args.batch_size
253253
eval_batch_size = args.eval_batch_size
254254
if use_torchacc():
255-
trian_batch_size *= world_size
255+
train_batch_size *= world_size
256256
eval_batch_size *= world_size
257-
training_args.per_device_train_batch_size = trian_batch_size
257+
training_args.per_device_train_batch_size = train_batch_size
258258
training_args.per_device_eval_batch_size = eval_batch_size
259259
training_args.group_by_length = use_torchacc()
260260

0 commit comments

Comments
 (0)