tao-finetune-cosmos-embed

Fine-tune and evaluate the Cosmos-Embed1 video-text embedding model.

83|20|Updated Apr 16, 2026
One-click install
npx skills add https://github.com/NVIDIA-TAO/tao-skill-bank --skill tao-finetune-cosmos-embed
Or copy as Structured Prompt for Agent
Please help me install this Agent Skill.
Skill: tao-finetune-cosmos-embed
Source: https://github.com/NVIDIA-TAO/tao-skill-bank/tree/main/skills/models/tao-finetune-cosmos-embed
Command: npx skills add https://github.com/NVIDIA-TAO/tao-skill-bank --skill tao-finetune-cosmos-embed

SYSTEM DOCUMENTATION & REQUIREMENTS

💡 This Skill requires docker, nvidia-container-toolkit, huggingface_token, and includes scripts (resource) and references (resource) and assets (resource) components.

What problem does it solve?

This Skill addresses the need to fine-tune and evaluate the Cosmos-Embed1 video-text embedding model for applications such as text-to-video retrieval, video-to-video search, semantic deduplication, and fine-tuning.

Core Features & Use Cases

  • Video-Text Embedding: Joint video-text embedder for a variety of tasks.
  • Fine-Tuning and Evaluation: Supports the training, evaluation, and export of fine-tuned models.
  • Use Case: For instance, fine-tuning the Cosmos-Embed1 model to improve text-to-video retrieval for a specific dataset or application.

Quick Start

Run the cosmos-embed1 train command with the appropriate configuration for your dataset.

Frequently Asked Questions about tao-finetune-cosmos-embed

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

FAQPage Schema
How do I fine-tune a video-text embedding model for text-to-video retrieval?

To fine-tune a video-text embedding model for text-to-video retrieval, run the `cosmos-embed1 train` command with your dataset configuration. This adapts the Cosmos-Embed1 model to improve retrieval accuracy for your specific application.

What is video-text embedding used for in semantic deduplication?

Video-text embedding maps joint video and text features into a shared space for semantic deduplication. It identifies duplicate video content by comparing embeddings, supporting text-to-video retrieval and video-to-video search applications.

Do I need Docker and nvidia-container-toolkit to run Cosmos-Embed1 fine-tuning?

Yes, you need Docker and nvidia-container-toolkit to run Cosmos-Embed1 fine-tuning. You also need a Huggingface token for model access. These dependencies provide the containerized GPU environment required for training and evaluation.

Can I evaluate and export a fine-tuned Cosmos-Embed1 model in the same workflow?

Yes, you can evaluate and export a fine-tuned Cosmos-Embed1 model in the same workflow. The Skill supports the complete training, evaluation, and export pipeline for customized video-text embedding models.

When should I use a fine-tuned video-text embedder instead of the base model?

You should use a fine-tuned video-text embedder instead of the base model when your application involves domain-specific text-to-video retrieval or semantic deduplication. Fine-tuning customizes the Cosmos-Embed1 model to recognize patterns unique to your dataset.