Skip to content

Commit

Permalink
fix: function name from requires_agumentation to requires_augmentation
Browse files Browse the repository at this point in the history
Signed-off-by: Will Johnson <[email protected]>
  • Loading branch information
willmj committed Jan 6, 2025
1 parent 3dc8ef7 commit d694e18
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion tuning/sft_trainer.py
Original file line number Diff line number Diff line change
Expand Up @@ -315,7 +315,7 @@ def train(
time.time() - data_preprocessing_time
)

if framework is not None and framework.requires_agumentation:
if framework is not None and framework.requires_augmentation:
model, (peft_config,) = framework.augmentation(
model, train_args, modifiable_args=(peft_config,)
)
Expand Down

0 comments on commit d694e18

Please sign in to comment.