Skip to main content

Python class

EmbeddingsGenerationInputs

EmbeddingsGenerationInputsโ€‹

class max.interfaces.EmbeddingsGenerationInputs(batches)

source

Bases: PipelineInputs

Batched inputs for an embeddings generation pipeline step.

Parameters:

batches (list[dict[RequestID, EmbeddingsContext]])

batchโ€‹

property batch: dict[RequestID, EmbeddingsContext]

source

Returns merged batches.

batchesโ€‹

batches: list[dict[RequestID, EmbeddingsContext]]

source