setup-benchmark-inputs

Prepare tokenized corpus shards and HF-DCP checkpoints for MoE benchmarks.

327|31|Updated Mar 21, 2026
One-click install
npx skills add https://github.com/mlc-ai/pith-train --skill setup-benchmark-inputs
Or copy as Structured Prompt for Agent
Please help me install this Agent Skill.
Skill: setup-benchmark-inputs
Source: https://github.com/mlc-ai/pith-train/tree/main/.agents/skills/setup-benchmark-inputs
Command: npx skills add https://github.com/mlc-ai/pith-train --skill setup-benchmark-inputs

SYSTEM DOCUMENTATION & REQUIREMENTS

💡 This Skill requires huggingface_hub, and includes scripts (resource) components.

What problem does it solve?

Set up the minimal artifacts needed to benchmark, profile, or regression-test a MoE model in PithTrain: a tokenized DCLM corpus shard and a converted HF checkpoint in DCP format. The process is idempotent and safe to re-run, ensuring outputs exist at workspace paths.

Core Features & Use Cases

  • Idempotent setup of benchmark artifacts: corpus shard tokenization and HF checkpoint conversion.
  • Supports multiple models (e.g., deepseek-v2-lite and qwen3-30b-a3b) with model-specific outputs under workspace.
  • Integrates with a local Python environment (.venv) and shell scripts to orchestrate fetch, tokenize, import, and conversion for reproducible benchmarking.

Quick Start

Run the setup script to generate the benchmark workspace for your chosen model.

Frequently Asked Questions about setup-benchmark-inputs

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

FAQPage Schema
How do I prepare a tokenized corpus shard and checkpoint for MoE benchmarking?

To prepare MoE benchmarking artifacts, you can automate tokenizing DCLM corpus shards and converting HF checkpoints to DCP format. This generates minimal reproducible workspace artifacts for profiling and regression testing.

What is the best way to set up PithTrain benchmark artifacts for a MoE model?

Setting up PithTrain benchmark artifacts involves orchestrating fetch, tokenize, import, and conversion steps via shell scripts. This idempotent process ensures required benchmark inputs exist at designated workspace paths without redundant reprocessing.

Can I re-run the MoE benchmark setup script if tokenized outputs already exist?

Yes, the MoE benchmark setup is idempotent and safe to re-run. It checks for existing tokenized corpus shards and converted HF checkpoints, automatically skipping steps when outputs already exist to prevent redundant processing.

Does the benchmark preparation workflow support models like deepseek-v2-lite and qwen3-30b-a3b?

Yes, the benchmark preparation supports multiple MoE models including deepseek-v2-lite and qwen3-30b-a3b. It generates model-specific outputs under the workspace, ensuring proper artifact configuration for each architecture.

Do I need huggingface_hub to convert checkpoints to DCP format for profiling?

Yes, huggingface_hub is required to fetch and convert HF checkpoints to DCP format. The setup integrates with a local Python environment to orchestrate the import and conversion needed for reproducible MoE profiling.

Why does my MoE regression testing require HF-DCP-formatted checkpoints?

HF-DCP-formatted checkpoints are required for MoE regression testing to ensure compatibility with PithTrain workflows. Converting standard HF checkpoints to DCP format provides a standardized input layout for consistent profiling and regression validation.