Skip to main content

Python function

max_tokens_to_generate

max_tokens_to_generate()

max.pipelines.lib.max_tokens_to_generate(prompt_size, max_length, max_new_tokens=None)

source

Returns the max number of new tokens to generate.

Parameters:

  • prompt_size (int)
  • max_length (int | None)
  • max_new_tokens (int | None)

Return type:

int | None