axolotl

Generate Axolotl YAML configurations for fine-tuning large language models.

247|22|Updated Dec 11, 2024
One-click install
npx skills add https://github.com/graniet/kheish --skill axolotl-graniet
Or copy as Structured Prompt for Agent▼
Please help me install this Agent Skill.
Skill: axolotl
Source: https://github.com/graniet/kheish/tree/main/skills/mlops/training/axolotl
Command: npx skills add https://github.com/graniet/kheish --skill axolotl-graniet

SYSTEM DOCUMENTATION & REQUIREMENTS

💡 This Skill includes references (resource) components.

What problem does it solve?

This skill eliminates the steep learning curve and configuration overhead of fine-tuning large language models with the Axolotl framework, which requires complex YAML setups for distributed training, LoRA adapters, and custom dataset formatting.

Core Features & Use Cases

  • Multi-Paradigm Fine-tuning Support: Covers supervised fine-tuning (SFT), preference-based post-training (DPO, KTO, ORPO, GRPO), and pre-training workflows for 100+ LLM architectures including Llama, Mistral, and multimodal models.
  • Production-Grade Configuration Guidance: Provides pre-validated YAML templates for common setups like single-GPU LoRA/QLoRA, multi-GPU DeepSpeed/FSDP distributed training, and mixed precision (FP16/BF16/FP8) training.
  • End-to-End Workflow Support: Includes guidance for dataset formatting (conversation, instruction, pre-tokenized), prompt strategy selection, inference, LoRA merging, and troubleshooting common training errors via an extensive FAQ.
  • Use Case: A machine learning engineer can use this skill to quickly configure a QLoRA fine-tuning job for a 7B parameter Llama model on a single consumer GPU, or scale to a 70B model fine-tuning job across 8 GPUs with DeepSpeed ZeRO-3 without manually writing hundreds of lines of error-prone configuration.

Quick Start

Provide your base model name, dataset path, and fine-tuning method (e.g. LoRA, DPO) to generate a ready-to-run Axolotl YAML config and launch your LLM fine-tuning job immediately.

Frequently Asked Questions about axolotl

High-intent search queries and answers about installing and using this skill.

FAQPage Schema
How do I generate an Axolotl YAML config for LoRA or QLoRA fine-tuning?▼

Generate a ready-to-run Axolotl YAML config for LoRA fine-tuning by providing your base model name, dataset path, and fine-tuning method to eliminate manual configuration complexity.

What is the best way to configure DeepSpeed or FSDP for distributed LLM training?▼

The best way to configure distributed LLM training is using pre-validated YAML templates for multi-GPU DeepSpeed and FSDP setups, avoiding error-prone manual configuration for large models.

Does Axolotl support preference-based post-training methods like DPO, KTO, and GRPO?▼

Yes, Axolotl supports preference-based post-training workflows including DPO, KTO, ORPO, and GRPO, alongside supervised fine-tuning and pre-training for over 100 open-source LLM architectures.

Can I fine-tune multimodal models using the Axolotl framework?▼

Yes, you can fine-tune multimodal models using the Axolotl framework, which provides configuration guidance and YAML templates specifically designed for multimodal architectures alongside text-only LLMs.

How do I format custom datasets for supervised fine-tuning in Axolotl?▼

Format custom datasets for supervised fine-tuning by following Axolotl's end-to-end dataset formatting guidance, which supports conversation, instruction, and pre-tokenized data structures for seamless training.

Why does my LLM fine-tuning job fail with common YAML configuration errors?▼

LLM fine-tuning jobs often fail due to YAML configuration errors, which you can resolve using the skill's extensive troubleshooting guidance for common training errors and pre-validated templates.