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 — 82 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 82 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.5 $0.00096 $0.02058
Sonnet 5.5 $0.00048 $0.01029
Haiku 4.5 $0.00024 $0.00514

Measured 1mo ago against content hash cb23a5ebfad4, method: parsed. Prices are Anthropic first-party input rates as of 2026-10-07, 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 1mo 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. 1mo 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 (97 stars, last pushed 5d ago), licensed Apache-2.0. It adds 241 tokens to every session and 5,144 once invoked, about $0.0010 per session on Opus 5.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

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

vss-benchmark-vlm-qa

Benchmark video Q&A accuracy and latency of a deployed RT-VLM (Cosmos Reason 3) via vss vlm run, using questions and videos from the DSS vss-devx-base dataset. Replaces the deprecated nat eval / vss-agent QA path. Not for tool-calling or trajectory evaluation, and not for LVS summarization throughput.

NVIDIA-AI-Blueprints/video-search-and-summarization · 80 tokens

rtvi-byom-porting

Use when adding, debugging, or validating a bring-your-own VLM in VSS RT-VLM, including a custom Hugging Face or NGC checkpoint loaded with configuration only (VLMMODELTOUSE=vllm-compatible), plus adapters, plugins, shims, and model-specific runtime dependencies. Not for ordinary catalog-preset deployment of an…

NVIDIA-AI-Blueprints/video-search-and-summarization · 93 tokens

profile-vllm-performance

Profile vLLM and RT-VLM inference to identify and remove GPU-idle gaps, underfilled batches, transfer stalls, serialized multimodal work, scheduler gaps, or KV pressure. Use this skill when GPU utilization is unexpectedly low or bursty, capacity trails another runtime, TTFT or ITL regresses, or multimodal prefill and…

NVIDIA-AI-Blueprints/video-search-and-summarization · 93 tokens

vss-build-vision-ai

Add agent-ready vision capabilities — dense captioning, detection, search, alerting, summarization — to an agent or application through a customizable, self-contained vision stack built on the NVIDIA VSS Blueprint. Use this skill when a developer or agent wants to give their app vision: pick capabilities via guided…

NVIDIA-AI-Blueprints/video-search-and-summarization · 203 tokens