Introduction

Axolotl offers a range of tools for fine-tuning language models (LLMs) with pre-trained weights and support frameworks like Hugging Face Transformers. Runpod is a scalable GPU cloud server provider that provides good environments for running machine learning workloads, which makes it a good option for high-resource LLM fine-tuning tasks. This tutorial will show how to set up Axolotl on Runpod to streamline LLM fine-tuning.

Prerequisites

To get the best out of this guide, you need specific resources and technical skills:

• A high-end GPU, a compatible OS, and Python 3.8 or higher.
• A Runpod account.
• Proficiency with basic Linux commands, python, and model-finetuning principles.

Setting Up the Environment on Runpod

Choosing a Runpod instance

When selecting your instance, you should meet your model’s demand and pick the accurate GPU, storage, and RAM. Running a 7B-parameter model on a single A100 with 40GB VRAM might scale but larger models like 13B or above will not scale on that same instance but a multi-GPU instance or an A100 with 80GB VRAM.

There’s an overview of the hourly cost involved in running each instance type on Runpod’s pricing page, and you can choose based on your workload requirements and budget.

If you'd like to skip the setup below, feel free to just deploy this axolotl template by winglian. If you'd rather install it from scratch, you can do that in any Pytorch pod.

Installing Axolotl and Setup

Environment Setup

Create a virtual environment for your project if you prefer:‍


Install axolotl

You can easily install Axolotl on the terminal from GitHub with the code below:

Data preparation for fine-tuning

Axoltol supports data in different formats like CSV, JSON, etc, so structuring the dataset to meet the training, validation, testing, and testing models is important.

Uploading data to Runpod

You can transfer the dataset to Runpod via SCP or use cloud storage like S3 and SFTP. For example, using SCP to transfer a dataset file:

If you are working with a small dataset, you could easily simply drag and drop it into the pod with Jupyter Notebook, or upload it using runpodctl.

Data formatting

Configuring Axolotl for fine-tuning

Creating a configuration file

Axolotl uses YAML configuration files. Create a file named config.yml with the following structure:

You might also look at the /examples/ folder for several premade .yml files that might also suit your needs.

Adjust parameters like base_model, model_type, lora_target_modules, and resources based on your specific model and hardware constraints.

Parameter explanation

• Efficient training methods:
  • load_in_8bit: true: Uses 8-bit quantization to reduce VRAM usage
  • adapter: lora: Uses LoRA adapter for parameter-efficient fine-tuning
  • lora_r, lora_alpha: Controls the rank and scaling of LoRA adapters
• Batch size and resources:
  • micro_batch_size: Size of each training batch
  • gradient_accumulation_steps: Accumulates gradients before updating weights
  • Adjust these based on your GPU memory

Running the fine-tuning process on Runpod

Start the fine-tuning process with:

For multi-GPU training with DeepSpeed:

Monitoring the training process

Monitor training progress directly in the terminal output. For more detailed monitoring:

1. Weights & Biases: If you've configured wandb_project in your config, you can monitor training metrics in real-time at wandb.ai.
2. TensorBoard: Axolotl saves logs that can be viewed with TensorBoard: --logdir ./output/tensorboard
3. GPU Monitoring: Use Runpod's dashboard or run: -n 1 nvidia-smi

Evaluating and exporting the fine-tuned model

Evaluating the model’s performance

Evaluate your model with Axolotl's built-in evaluation:

Conclusion

To maximize the efficiency and minimize the costs on Runpod:

1. Select the perfect instance size for your model
2. Use LoRA and quantization techniques to reduce VRAM requirements
3. Utilize Runpod volumes for data persistence between sessions
4. Monitor training actively with W&B or TensorBoard
5. Consider spot instances for non-critical training jobs to reduce costs

For hyperparameter tuning, experiment with different learning rates, LoRA configurations, and batch sizes while monitoring the model's performance.

‍

Author profile: James Sandy