> For clean Markdown of any page, append .md to the page URL.
> For a complete documentation index, see https://docs.nvidia.com/nemo/curator/llms.txt.
> For AI client integration (Claude Code, Cursor, etc.), connect to the MCP server at https://docs.nvidia.com/nemo/curator/_mcp/server.

# nemo_curator.stages.image.embedders.clip_embedder

## Module Contents

### Classes

| Name                                                                                            | Description                                             |
| ----------------------------------------------------------------------------------------------- | ------------------------------------------------------- |
| [`ImageEmbeddingStage`](#nemo_curator-stages-image-embedders-clip_embedder-ImageEmbeddingStage) | Stage for generating image embeddings using CLIP model. |

### API

```python
class nemo_curator.stages.image.embedders.clip_embedder.ImageEmbeddingStage(
    model_dir: str = None,
    num_gpus_per_worker: float = 0.25,
    model_inference_batch_size: int = 32,
    verbose: bool = False,
    remove_image_data: bool = False,
    name: str = 'image_embedding'
)
```

Dataclass

**Bases:** [ProcessingStage\[ImageBatch, ImageBatch\]](/nemo-curator/nemo_curator/stages/base#nemo_curator-stages-base-ProcessingStage)

Stage for generating image embeddings using CLIP model.

This class processes image batches through a CLIP model to generate
embeddings for each image. It assumes image data is already loaded
in ImageObject.image\_data and stores embeddings in ImageObject.embedding.

**`model_dir`** `str = None`

---

**`model_inference_batch_size`** `int = 32`

---

**`name`** `str = 'image_embedding'`

---

**`num_gpus_per_worker`** `float = 0.25`

---

**`remove_image_data`** `bool = False`

---

**`verbose`** `bool = False`

---

```python
nemo_curator.stages.image.embedders.clip_embedder.ImageEmbeddingStage.__post_init__() -> None
```

```python
nemo_curator.stages.image.embedders.clip_embedder.ImageEmbeddingStage.inputs() -> tuple[list[str], list[str]]
```

```python
nemo_curator.stages.image.embedders.clip_embedder.ImageEmbeddingStage.outputs() -> tuple[list[str], list[str]]
```

```python
nemo_curator.stages.image.embedders.clip_embedder.ImageEmbeddingStage.process(
    task: nemo_curator.tasks.ImageBatch
) -> nemo_curator.tasks.ImageBatch
```

Process an image batch to generate embeddings.

**Parameters:**

**`task`** `ImageBatch`

ImageBatch containing list of ImageObject instances with pre-loaded image\_data

---

**Returns:** `ImageBatch`

ImageBatch with embeddings stored in ImageObject.embedding

```python
nemo_curator.stages.image.embedders.clip_embedder.ImageEmbeddingStage.setup(
    _worker_metadata: nemo_curator.backends.base.WorkerMetadata | None = None
) -> None
```

Initialize the CLIP image embedding model.

```python
nemo_curator.stages.image.embedders.clip_embedder.ImageEmbeddingStage.setup_on_node(
    node_info: nemo_curator.backends.base.NodeInfo,
    worker_metadata: nemo_curator.backends.base.WorkerMetadata
) -> None
```

Download the weights for the CLIP model on the node.

```python
nemo_curator.stages.image.embedders.clip_embedder.ImageEmbeddingStage.yield_next_batch(
    task: nemo_curator.tasks.ImageBatch
) -> collections.abc.Generator[nemo_curator.tasks.ImageBatch, None, None]
```

Yield batches of images from the task.

**Parameters:**

**`task`** `ImageBatch`

ImageBatch containing list of ImageObject instances with pre-loaded image\_data

---