Upload ai-ml/hf-finetuning/finetune.py with huggingface_hub

ai-ml/hf-finetuning/finetune.py

@@ -149,7 +149,7 @@ def finetune(config: FinetuneConfig):
         per_device_train_batch_size=config.per_device_train_batch_size,
         gradient_accumulation_steps=config.gradient_accumulation_steps,
         learning_rate=config.learning_rate,
-        max_seq_length=config.max_seq_length,
+        max_length=config.max_seq_length,
         warmup_ratio=config.warmup_ratio,
         lr_scheduler_type=config.lr_scheduler_type,
         bf16=config.bf16,