alpamayo1-sft

Automate end-to-end fine-tuning of the Alpamayo-1 VLM on the PAI dataset.

108|25|Updated Apr 22, 2026
One-click install
npx skills add https://github.com/NVlabs/alpamayo-recipes --skill alpamayo1-sft
Or copy as Structured Prompt for Agent
Please help me install this Agent Skill.
Skill: alpamayo1-sft
Source: https://github.com/NVlabs/alpamayo-recipes/tree/main/recipes/alpamayo1_sft
Command: npx skills add https://github.com/NVlabs/alpamayo-recipes --skill alpamayo1-sft

SYSTEM DOCUMENTATION & REQUIREMENTS

💡 This Skill requires deepspeed, transformers, torch, torchvision, alpamo1_r1, alpamo1-recipes, flash-attn, hydra-core, torchmetrics, pandas, opencv-python, physical_ai_av, wandb, deepspeed, and includes scripts (resource) and references (resource) and assets (resource) components.

What problem does it solve?

This Skill automates the end-to-end supervised fine-tuning process for the Alpamayo-1 Vision-Learning Model (VLM) on the Physical AI Autonomous Vehicles (PAI) dataset. It removes the need for manual fine-tuning steps and speeds up the process.

Core Features & Use Cases

  • End-to-End Fine-Tuning: Automates the entire fine-tuning process, including Stage 1 and Stage 2, using Hugging Face, Hydra, and DeepSpeed.
  • Data Preparation: Provides commands to download and prepare the PAI dataset, including options for selecting a subset or enabling CoC reasoning labels.
  • Model Training: Handles the installation and configuration of the necessary environment, including the uv virtual environment and the required libraries.
  • Evaluation: Includes commands for evaluating the trained model on the validation dataset.
  • Use Case: Ideal for researchers or developers looking to fine-tune the Alpamayo-1 VLM on the PAI dataset without manually setting up the environment or training pipeline.

Quick Start

Use the alpamayo1-sft skill to fine-tune the Alpamayo-1 VLM on the PAI dataset using the default configuration.

Frequently Asked Questions about alpamayo1-sft

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

FAQPage Schema
How do I fine-tune the Alpamayo-1 VLM on the Physical AI Autonomous Vehicles dataset?

You can fine-tune the Alpamayo-1 VLM on the PAI dataset by running an automated end-to-end pipeline that handles data preparation, Stage 1 and Stage 2 training, and model evaluation.

Do I need DeepSpeed and Hugging Face transformers to train the Alpamayo-1 VLM?

Yes, you need DeepSpeed and Hugging Face transformers to train the Alpamayo-1 VLM. The supervised fine-tuning process also requires Hydra for configuration and flash-attn for model training.

What is the best way to automate VLM fine-tuning for autonomous vehicle datasets?

The best way to automate VLM fine-tuning for autonomous vehicle datasets is using a pipeline that manages environment setup, data preparation, and multi-stage training without manual intervention.

Can I select a subset of the PAI dataset for supervised VLM fine-tuning?

Yes, you can select a subset of the PAI dataset for supervised VLM fine-tuning. The data preparation commands include options for subsetting data and enabling CoC reasoning labels.

How does Hydra configuration work with DeepSpeed for vision-learning model training?

Hydra configuration works with DeepSpeed for vision-learning model training by orchestrating the multi-stage supervised fine-tuning parameters and environment variables automatically.

Why does VLM fine-tuning require a specific virtual environment setup?

VLM fine-tuning requires a specific virtual environment setup to manage complex dependencies like flash-attn, DeepSpeed, and torchmetrics, ensuring the training pipeline executes without version conflicts.