ad-model-onboard

Translate HuggingFace models into prefill-only AutoDeploy custom models with equivalence tests.

14.3k|2.6k|Updated Aug 16, 2023
One-click install
npx skills add https://github.com/NVIDIA/TensorRT-LLM --skill ad-model-onboard
Or copy as Structured Prompt for Agent
Please help me install this Agent Skill.
Skill: ad-model-onboard
Source: https://github.com/NVIDIA/TensorRT-LLM/tree/main/.claude/skills/ad-model-onboard
Command: npx skills add https://github.com/NVIDIA/TensorRT-LLM --skill ad-model-onboard

SYSTEM DOCUMENTATION & REQUIREMENTS

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

What problem does it solve?

This Skill automates the complex process of converting HuggingFace models into a prefill-only AutoDeploy custom model format, ensuring compatibility and efficient deployment.

Core Features & Use Cases

  • Model Translation: Converts HuggingFace models to a prefill-only AutoDeploy format.
  • Custom Op Integration: Utilizes reference custom operations for compatibility.
  • Validation: Performs hierarchical equivalence tests to ensure accuracy.
  • Use Case: A machine learning engineer needs to deploy a new LLM from HuggingFace to a custom inference engine. This Skill handles the intricate conversion and validation steps, saving significant development time.

Quick Start

Use the ad-model-onboard skill to convert the HuggingFace model 'mistralai/Mistral-7B-v0.1' into a prefill-only AutoDeploy custom model.

Frequently Asked Questions about ad-model-onboard

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

FAQPage Schema
How do I convert HuggingFace models for AutoDeploy inference?

You convert HuggingFace models for AutoDeploy inference by translating them into a prefill-only custom model format using reference custom operations. This process ensures the LLM is fully compatible with the custom inference engine.

What is a prefill-only custom model format in LLM inference optimization?

A prefill-only custom model format is a specialized deployment structure for LLM inference. It is generated by translating HuggingFace architectures into AutoDeploy custom models using specific reference custom operations for efficient inference.

How do I validate model accuracy after converting HuggingFace weights?

You validate model accuracy after converting HuggingFace weights by performing hierarchical equivalence tests. This method rigorously compares the translated AutoDeploy custom model against the original HuggingFace reference for numerical equivalence.

Can I integrate custom operations when onboarding HuggingFace LLMs?

Yes, you can integrate custom operations when onboarding HuggingFace LLMs. The conversion process specifically utilizes reference custom operations to ensure the translated prefill-only AutoDeploy model maintains architectural compatibility.

What is the best way to deploy a new LLM from HuggingFace to a custom inference engine?

The best way to deploy a new LLM from HuggingFace to a custom inference engine is automating its conversion into a prefill-only AutoDeploy format. This approach handles intricate translation and hierarchical validation, saving significant development time.

Does converting HuggingFace models to AutoDeploy format change model architecture?

Converting HuggingFace models to AutoDeploy format translates the model architecture into a prefill-only structure. It uses reference custom operations and validates the integration through rigorous numerical comparisons against the original HuggingFace reference.