Skip to main content

Python class

PipelineTask

PipelineTask

class max.interfaces.PipelineTask(value, names=<not given>, *values, module=None, qualname=None, type=None, start=1, boundary=None)

source

Bases: str, Enum

Enum representing the types of pipeline tasks supported.

AUDIO_GENERATION

AUDIO_GENERATION = 'audio_generation'

source

Task for generating audio.

EMBEDDINGS_GENERATION

EMBEDDINGS_GENERATION = 'embeddings_generation'

source

Task for generating embeddings.

PIXEL_GENERATION

PIXEL_GENERATION = 'pixel_generation'

source

Task for generating pixels.

SPEECH_TOKEN_GENERATION

SPEECH_TOKEN_GENERATION = 'speech_token_generation'

source

Task for generating speech tokens.

TEXT_GENERATION

TEXT_GENERATION = 'text_generation'

source

Task for generating text.

UNDEFINED

UNDEFINED = 'undefined'

source

Undefined task, used as default when task should be auto-detected.

output_type

property output_type: type[dict[RequestID, SchedulerResult[Any]]]

source

Get the output type for the pipeline task.

Returns:

The output type for the pipeline task.