Class ModelParams.Builder

  • Enclosing class:
    ModelParams

    public static class ModelParams.Builder
    extends Object
    • Constructor Detail

      • Builder

        public Builder()
    • Method Detail

      • maxTokens

        public ModelParams.Builder maxTokens​(Integer maxTokens)
        Maximum number of tokens.
        Parameters:
        maxTokens - the value to set
        Returns:
        this builder
      • temperature

        public ModelParams.Builder temperature​(Float temperature)
        Sampling temperature.
        Parameters:
        temperature - the value to set
        Returns:
        this builder
      • topP

        public ModelParams.Builder topP​(Float topP)
        Nucleus sampling probability.
        Parameters:
        topP - the value to set
        Returns:
        this builder
      • topK

        public ModelParams.Builder topK​(Integer topK)
        Top-K sampling parameter.
        Parameters:
        topK - the value to set
        Returns:
        this builder
      • frequencyPenalty

        public ModelParams.Builder frequencyPenalty​(Float frequencyPenalty)
        Frequency penalty to reduce repeating tokens.
        Parameters:
        frequencyPenalty - the value to set
        Returns:
        this builder
      • presencePenalty

        public ModelParams.Builder presencePenalty​(Float presencePenalty)
        Presence penalty to encourage new topics.
        Parameters:
        presencePenalty - the value to set
        Returns:
        this builder