tao-finetune-huggingface-model

tao-finetune-huggingface-model is a skill for Claude Code from NVIDIA-TAO/tao-skill-bank. It costs 241 tokens per session (5,144 once invoked), scanned C, original, Apache-2.0.

A workflow for training Hugging Face computer-vision, vision-language, or language models on a local NVIDIA GPU, including full training or LoRA fine-tuning.

In plain words
What is it for?
Use it to prepare a training pipeline, run a small test, fine-tune a model, and publish the resulting model to Hugging Face Hub.
Why use it?
It provides a defined training process in an NVIDIA container when no model-specific workflow applies, helping make experiments repeatable.

Skill for Claude Code

Written for Claude Code: allowed-tools in frontmatter.

Part of the tao-skills plugin — 76 skills shipped together , and of tao-skill-bank

Good fit Use it to prepare a training pipeline, run a small test, fine-tune a model, and publish the resulting model to Hugging Face Hub.

Compare 6 skills from other repositories ↓
Install with agentmods
npx agentmods add skills/nvidia-tao/tao-skill-bank/tao-finetune-huggingface-model
Install

Getting it into your agent

One page per mod, every tool's command on it. A separate URL per tool would split the same page into five that compete with each other.

Any agent
npx skills add NVIDIA-TAO/tao-skill-bank --skill tao-finetune-huggingface-model
Clone the repo
git clone --depth 1 https://github.com/NVIDIA-TAO/tao-skill-bank

Made for: Claude Code.

Or install tao-skills, the plugin that ships this one along with the rest of its 76 skills.

Wrote this? Show the measurements

A badge with what this costs and how it scanned, read live from this page, so it follows the numbers instead of freezing them. Markdown for a README, HTML for a documentation site or a project page.

agentmods badge for tao-finetune-huggingface-model

README.md
[![agentmods](https://agentmods.dev/badge/skills/nvidia-tao/tao-skill-bank/tao-finetune-huggingface-model/github.svg)](https://agentmods.dev/skills/nvidia-tao/tao-skill-bank/tao-finetune-huggingface-model)
Your own site
<a href="https://agentmods.dev/skills/nvidia-tao/tao-skill-bank/tao-finetune-huggingface-model"><img src="https://agentmods.dev/badge/skills/nvidia-tao/tao-skill-bank/tao-finetune-huggingface-model/github.svg" alt="Measured on agentmods" height="20"></a>

Or the 80×15 button, for a site that already has a row of RSS and ATOM ones. Only the verdict fits; the numbers stay here.

agentmods 80×15 button for tao-finetune-huggingface-model

Your own site · 80×15
<a href="https://agentmods.dev/skills/nvidia-tao/tao-skill-bank/tao-finetune-huggingface-model"><img src="https://agentmods.dev/badge/skills/nvidia-tao/tao-skill-bank/tao-finetune-huggingface-model.svg" alt="Reviewed on agentmods" width="80" height="20"></a>
Per session 241 Skills are progressive disclosure: only the name and description are preloaded; the body loads when the skill is used.
When invoked 5,144 The whole file, excluding the scripts and references it only reads on demand.
Security scan C 1 finding. A grade says what 26 rules found in the file — not that it is safe.
Origin original No closer match found in the catalogue.
Token cost

What it costs to keep this loaded

Counted locally with the o200k_base tokenizer, which is exact for GPT models; Claude uses its own tokenizer and its counts differ. Treat this as one consistent yardstick across the catalogue rather than a bill. Prices are per million input tokens.

ModelPer sessionOnce invoked
Fable 5.1 $0.00241 $0.05144
Opus 5 $0.00120 $0.02572
Sonnet 5 $0.00048 $0.01029
Haiku 4.5 $0.00024 $0.00514

Measured 12d ago against content hash cb23a5ebfad4, method: parsed. Prices are Anthropic first-party input rates as of 2026-09-12, from the pricing page.

Security

Grade C, and why

tao-finetune-huggingface-model scanned grade C with 1 finding against 26 rules in 11 categories — prompt injection, anti-refusal, data exfiltration, privilege escalation, supply chain, agent snooping, system-prompt leakage, SSRF and excessive agency — measured 12d ago.

A static scan of the body, not an audit. Every finding is printed with the line that produced it so you can judge whether it matters here. A mod is markdown that instructs an agent; that is exactly why what it instructs is worth reading.

Recursive force deletehighDestructive command

rm -rf with a variable or a broad path is one typo away from removing the wrong tree.

`references/workflow-intake-preflight.md`). Optionally `rm -rf "$OUTPUT_DIR/.probe"`
skills/applications/tao-finetune-huggingface-model/SKILL.md · 405 lines

How it starts

The opening of the file, as written. The whole thing — 405 lines — stays where its author put it; the contents beside it link to each section on GitHub.

tao-finetune-huggingface-model

Standalone install? If this session was not initialized by the TAO skill bank plugin, run the tao-setup skill first (host preflight, credentials, cross-skill discovery).

Local NVIDIA GPU fine-tuning for HuggingFace models, grounded in live-fetched documentation with curated references as a fallback safety net. One NGC container, a few focused scripts, one push to HF Hub. Follow the rules in this file; don't improvise.

Dedicated-model routing gate

Before Step 1 or any probe, image selection, package install, venv creation, or training-code generation, resolve model_id against the packaged model-owner registry. Use the absolute skill-bank root from which this file was loaded:

python <bank-root>/scripts/resolve_tao_model.py \
  --skill-bank <bank-root> \
  --model "$MODEL_ID" \
  --format json

The resolver matches model metadata, including huggingface_model_ids, network_arch, skill names, and legacy aliases. Routing is internal: a model ID and task are enough. Never require prompt boilerplate about skills, containers, or checkpoint formats.

  • Exit 0: stop this workflow and follow the owning model skill's environment, action metadata, preflight, and checkpoint preparation.
  • Exit 3: no packaged model skill owns the ID. This is the only result that permits Step 1 of the generic workflow.
  • Any other nonzero exit: ownership discovery is broken or ambiguous. Stop and resolve that error; do not silently fall back to generic Hugging Face training.

Hugging Face hosting never overrides ownership. Do not use this workflow to bypass a matched skill or ask the user to prescribe its internal preparation. For example, nvidia/Cosmos3-Nano routes to tao-finetune-cosmos-reason.

Do not create a host training venv in this workflow. Its default execution path is the NGC container documented below; any venv-based training path requires an explicit user request.

Read the full file on GitHub · 405 lines

Changes

What this file has done since we first saw it

Hashed on every crawl. A supply-chain change to an agent config is a question of when, not whether, so the history is kept rather than the latest state alone.

  1. 12d ago First seen · 405 lines · 241 tokens per session scan C cb23a5ebfad4

Subscribe to this mod's changes

tao-finetune-huggingface-model is a skill published in the GitHub repository NVIDIA-TAO/tao-skill-bank (88 stars, last pushed today), licensed Apache-2.0. It adds 241 tokens to every session and 5,144 once invoked, about $0.0012 per session on Opus 5. A static security scan graded it C with 1 finding (recursive force delete). No closer match exists in the catalogue, so it is treated as the original; first seen 2026-08-30.

Related

Other skills, from other repositories

ml-training

Machine learning model training authority — classifier and regressor training with scikit-learn and PyTorch, feature engineering, cross-validation, hyperparameter tuning, fine-tuning transformer models with HuggingFace, dataset splits, loss functions, learning rate schedules, and reproducible training runs.

LuuOW/meridian-mcp · 58 tokens

quantizing-models-bitsandbytes

Quantizes LLMs to 8-bit or 4-bit for 50-75% memory reduction with minimal accuracy loss. Use when GPU memory is limited, need to fit larger models, or want faster inference. Supports INT8, NF4, FP4 formats, QLoRA training, and 8-bit optimizers. Works with HuggingFace Transformers.

davila7/claude-code-templates · 83 tokens

nanogpt

Educational GPT implementation in 300 lines. Reproduces GPT-2 (124M) on OpenWebText. Clean, hackable code for learning transformers. By Andrej Karpathy. Perfect for understanding GPT architecture from scratch. Train on Shakespeare (CPU) or OpenWebText (multi-GPU).

davila7/claude-code-templates · 65 tokens

llama-factory

Expert guidance for fine-tuning LLMs with LLaMA-Factory - WebUI no-code, 100+ models, 2/3/4/5/6/8-bit QLoRA, multimodal support.

davila7/claude-code-templates · 51 tokens

mlflow

Track ML experiments, manage model registry with versioning, deploy models to production, and reproduce experiments with MLflow - framework-agnostic ML lifecycle platform.

davila7/claude-code-templates · 33 tokens

ml-training-recipes

Battle-tested PyTorch training recipes for all domains — LLMs, vision, diffusion, medical imaging, protein/drug discovery, spatial omics, genomics. Covers training loops, optimizer selection (AdamW, Muon), LR scheduling, mixed precision, debugging, and systematic experimentation. Use when training or fine-tuning…

Orchestra-Research/AI-Research-SKILLs · 88 tokens