For AI agents: a documentation index is available at the root level at /llms.txt and /llms-full.txt. Append /llms.txt to any URL for a page-level index, or .md for the markdown version of any page.
DocumentationAPI Reference
DocumentationAPI Reference
  • API Reference
    • Overview
        • Nemo Curator
          • Backends
          • Config
          • Core
          • Metrics
          • Models
            • Aesthetics
            • Base
            • Client
            • Clip
            • Cosmos Embed1
            • Nemotron 3 Nano Omni
            • Nemotron H Vl
            • Nsfw
            • Prompt Formatter
            • Qwen Lm
            • Qwen Vl
            • Transnetv2
            • Vllm Model
          • Package Info
          • Pipeline
          • Stages
          • Tasks
          • Utils
    • Pipeline
    • ProcessingStage
    • CompositeStage
    • Resources
NVIDIANVIDIA
Developer-friendly docs for your API
Privacy Policy | Your Privacy Choices | Terms of Service | Accessibility | Corporate Policies | Product Security | Contact

Copyright © 2026, NVIDIA Corporation.

LogoLogoNeMo Curator
On this page
  • Module Contents
  • Classes
  • API
API ReferenceFull Library ReferenceNemo CuratorNemo CuratorModels

nemo_curator.models.base

||View as Markdown|
Previous

nemo_curator.models.aesthetics

Next

nemo_curator.models.client

Module Contents

Classes

NameDescription
ModelInterfaceAbstract base class that defines an interface for machine learning models.

API

class nemo_curator.models.base.ModelInterface()
Abstract

Bases: ABC

Abstract base class that defines an interface for machine learning models.

Specifically focused on their weight handling and environmental setup.

This interface allows our pipeline code to download weights locally and setup models in a uniform way. It does not place any restrictions on how inference is run.

model_id_names
list[str]

Returns a list of model IDs associated with the model.

In cosmos-curate, each model has an ID associated with it. This is often the huggingspace name for that model (e.g. Salesforce/instructblip-vicuna-13b).

nemo_curator.models.base.ModelInterface.setup() -> None
abstract

Set up the model for use, such as loading weights and building computation graphs.