diff --git a/llama_stack/providers/inline/post_training/huggingface/recipes/finetune_single_device.py b/llama_stack/providers/inline/post_training/huggingface/recipes/finetune_single_device.py index ed9cd7755..2a024eb25 100644 --- a/llama_stack/providers/inline/post_training/huggingface/recipes/finetune_single_device.py +++ b/llama_stack/providers/inline/post_training/huggingface/recipes/finetune_single_device.py @@ -469,7 +469,7 @@ class HFFinetuningSingleDevice: use_cpu=True if device.type == "cpu" and not torch.backends.mps.is_available() else False, save_strategy=save_strategy, report_to="none", - max_seq_length=provider_config.max_seq_length, + max_length=provider_config.max_seq_length, gradient_accumulation_steps=config.gradient_accumulation_steps, gradient_checkpointing=provider_config.gradient_checkpointing, learning_rate=lr,