gptj

Documentation

Index

• type GptJ
  • func Load(path string) *GptJ
  • func (g *GptJ) Prompt(prompt string, po PromptOptions) string
• type PromptOptions

Types

type GptJ

type GptJ struct {
	// contains filtered or unexported fields
}

func Load

func Load(path string) *GptJ

func (*GptJ) Prompt

func (g *GptJ) Prompt(prompt string, po PromptOptions) string

type PromptOptions

type PromptOptions struct {
	Tokens int
	TopK   int
	TopP   float64
	Temp   float64
	Batch  int
}