Interface InferenceConfiguration.Builder

  • Method Details

    • maxTokens

      The maximum number of tokens to generate.

      Parameters:
      maxTokens - The maximum number of tokens to generate.
      Returns:
      Returns a reference to this object so that method calls can be chained together.
    • temperature

      InferenceConfiguration.Builder temperature(Float temperature)

      The temperature for sampling.

      Parameters:
      temperature - The temperature for sampling.
      Returns:
      Returns a reference to this object so that method calls can be chained together.
    • topP

      The top-p value for nucleus sampling.

      Parameters:
      topP - The top-p value for nucleus sampling.
      Returns:
      Returns a reference to this object so that method calls can be chained together.
    • stopSequences

      InferenceConfiguration.Builder stopSequences(Collection<String> stopSequences)

      Stop sequences that end generation.

      Parameters:
      stopSequences - Stop sequences that end generation.
      Returns:
      Returns a reference to this object so that method calls can be chained together.
    • stopSequences

      InferenceConfiguration.Builder stopSequences(String... stopSequences)

      Stop sequences that end generation.

      Parameters:
      stopSequences - Stop sequences that end generation.
      Returns:
      Returns a reference to this object so that method calls can be chained together.