Lightlines
By nvidia

tao-finetune-cosmos-embed

nvidia1.7k installs

Find matching videos with text or other videos.

What it does

Fine-tunes, evaluates, runs, and exports Cosmos-Embed1 models for finding videos by text or video similarity.

When to use it

Use when asked to fine-tune, run inference with, export, or embed and search videos using Cosmos-Embed1.

How to use it

Give it videos, metadata, and the requested model operation; it runs Cosmos-Embed1 and saves checkpoints, evaluation metrics, search results, or exported models locally.

What you provide

  • Your files

Uses

Security reviews

Not yet reviewed.


What you need · 5

Requires Docker to pull and run the published Cosmos-Embed container.

Requires NVIDIA Container Toolkit for the GPU-enabled container workflow.


Access · 3

This skill

Hugging Face

Read

Downloads pretrained model components

nvcr.io/nvidia/tao/tao-toolkit:7.3.0-cosmos-embed

Execute

Pulls the Cosmos-Embed container