Skip to main content

/

Docs

Nightly

v24.6

Log in

Python module

tokenizer

Implementations of provided tokenizers.

IdentityPipelineTokenizer​

class max.pipelines.tokenizer.IdentityPipelineTokenizer(*args, **kwargs)

decode()​

async decode(context: TokenGeneratorContext, encoded: Any) → str

Decodes response tokens to text.

  • Parameters:

    • context (TokenGeneratorContext) – Current generation context.
    • encoded (TokenizerEncoded) – Encoded response tokens.

  • Returns:

    Un-encoded response text.

  • Return type:

    str

encode()​

async encode(prompt: str) → str

Encodes text prompts as tokens.

  • Parameters:

    prompt (str) – Un-encoded prompt text.

  • Raises:

    ValueError – If the prompt exceeds the configured maximum length.

  • Returns:

    Encoded prompt tokens.

  • Return type:

    TokenizerEncoded

eos​

property eos*: int*

PreTrainedPipelineTokenizer​

class max.pipelines.tokenizer.PreTrainedPipelineTokenizer(delegate: PreTrainedTokenizer | PreTrainedTokenizerFast)

apply_chat_template()​

apply_chat_template(messages: list[max.pipelines.interfaces.TokenGeneratorRequestMessage]) → str

decode()​

async decode(context: TokenGeneratorContext, encoded: ndarray) → str

Decodes response tokens to text.

  • Parameters:

    • context (TokenGeneratorContext) – Current generation context.
    • encoded (TokenizerEncoded) – Encoded response tokens.

  • Returns:

    Un-encoded response text.

  • Return type:

    str

encode()​

async encode(prompt: str) → ndarray

Encodes text prompts as tokens.

  • Parameters:

    prompt (str) – Un-encoded prompt text.

  • Raises:

    ValueError – If the prompt exceeds the configured maximum length.

  • Returns:

    Encoded prompt tokens.

  • Return type:

    TokenizerEncoded

eos​

property eos*: int*

TextAndVisionTokenizer​

class max.pipelines.tokenizer.TextAndVisionTokenizer(config: PipelineConfig)

Encapsulates creation of TextContext and specific token encode/decode logic.

apply_chat_template()​

apply_chat_template(messages: list[max.pipelines.interfaces.TokenGeneratorRequestMessage]) → str

decode()​

async decode(context: TextAndVisionContext, encoded: ndarray, **kwargs) → str

Transformer a provided encoded token array, back into readable text.

encode()​

async encode(prompt: str) → ndarray

Transform the provided prompt into a token array.

eos​

property eos*: int*

new_context()​

async new_context(request: TokenGeneratorRequest) → TextAndVisionContext

Create a new TextAndVisionContext object, leveraging necessary information like cache_seq_id and prompt from TokenGeneratorRequest.

TextTokenizer​

class max.pipelines.tokenizer.TextTokenizer(config: PipelineConfig, enable_llama_whitespace_fix: bool = False)

Encapsulates creation of TextContext and specific token encode/decode logic.

apply_chat_template()​

apply_chat_template(messages: list[max.pipelines.interfaces.TokenGeneratorRequestMessage]) → str

decode()​

async decode(context: TextContext, encoded: ndarray, **kwargs) → str

Transformer a provided encoded token array, back into readable text.

encode()​

async encode(prompt: str) → ndarray

Transform the provided prompt into a token array.

eos​

property eos*: int*

new_context()​

async new_context(request: TokenGeneratorRequest) → TextContext

Create a new TextContext object, leveraging necessary information like cache_seq_id and prompt from TokenGeneratorRequest.

max_tokens_to_generate()​

max.pipelines.tokenizer.max_tokens_to_generate(prompt_size: int, max_length: int, max_new_tokens: int = -1) → int

Returns the max number of new tokens to generate.

run_with_default_executor()​

async max.pipelines.tokenizer.run_with_default_executor(fn, *args)