adding-model-support

Add language or multimodal models to Megatron-Bridge via config analysis and custom weight mapping.

852|445|Updated May 21, 2025
One-click install
npx skills add https://github.com/NVIDIA-NeMo/Megatron-Bridge --skill adding-model-support-nvidia-nemo
Or copy as Structured Prompt for Agent
Please help me install this Agent Skill.
Skill: adding-model-support
Source: https://github.com/NVIDIA-NeMo/Megatron-Bridge/tree/main/skills/adding-model-support
Command: npx skills add https://github.com/NVIDIA-NeMo/Megatron-Bridge --skill adding-model-support-nvidia-nemo

SYSTEM DOCUMENTATION & REQUIREMENTS

💡 This Skill includes references (resource) and scripts (resource) components.

What problem does it solve?

This Skill provides comprehensive guidance for incorporating new language or vision-language models into the Megatron-Bridge framework, simplifying the process of supporting diverse architectures.

Core Features & Use Cases

  • Model Analysis and Configuration: Guides users through analyzing HuggingFace model configs, identifying custom layers, and handling quantization.
  • Bridge and Provider Implementation: Assists in developing model-specific code for weight mapping, configuration translation, and support scripts.
  • Use Case: For example, onboard a new multilingual LLM or multimodal VLM by analyzing its configs, creating mapping classes, and establishing testing pipelines.

Quick Start

Use the guide to analyze the model configs and set up the correct bridge, mapping, and test scripts for seamless integration.

Frequently Asked Questions about adding-model-support

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

FAQPage Schema
How do I add a new HuggingFace model to Megatron-Bridge?

To add a new HuggingFace model to Megatron-Bridge, you analyze the model configs, create custom weight mapping classes, and implement configuration translation. This Skill guides you through establishing the bridge, mapping, and test scripts for seamless integration.

What is the process for multimodal model integration with Megatron conversion tooling?

Multimodal model integration with Megatron conversion tooling involves analyzing vision-language configs, identifying custom layers, and handling quantization. This Skill provides implementation patterns to ensure compatibility and efficient deployment of diverse architectures.

Do I need prior experience with HuggingFace configs to use this model integration framework?

Yes, you need familiarity with HuggingFace configs and Megatron conversion tooling to use this model integration framework. The Skill requires this prerequisite knowledge to facilitate config analysis, custom weight mapping, and architecture validation.

What's the best way to handle custom weight mapping when onboarding a new LLM?

The best way to handle custom weight mapping when onboarding a new LLM is to develop model-specific code for configuration translation. This Skill assists in creating mapping classes and establishing testing pipelines for accurate weight conversion.

Does this approach support both language and vision-language model architectures?

Yes, this approach supports both language and vision-language model architectures. It provides comprehensive guidance for incorporating diverse architectures into Megatron-Bridge, ensuring compatibility and efficient validation across multimodal configurations.