finetune_model
Optimize model fine-tuning using Unsloth's efficient training methods to reduce memory usage and accelerate the process with specific parameters.
Instructions
Fine-tune a model with Unsloth optimizations
Input Schema
| Name | Required | Description | Default |
|---|---|---|---|
| model_name | Yes | Name of the model to fine-tune | |
| dataset_name | Yes | Name of the dataset to use for fine-tuning | |
| output_dir | Yes | Directory to save the fine-tuned model | |
| max_seq_length | No | Maximum sequence length for training | |
| lora_rank | No | Rank for LoRA fine-tuning | |
| lora_alpha | No | Alpha for LoRA fine-tuning | |
| batch_size | No | Batch size for training | |
| gradient_accumulation_steps | No | Number of gradient accumulation steps | |
| learning_rate | No | Learning rate for training | |
| max_steps | No | Maximum number of training steps | |
| dataset_text_field | No | Field in the dataset containing the text | |
| load_in_4bit | No | Whether to use 4-bit quantization |