utc

type BasePrompt struct {
	Position  int
	TokenType int
	Truncate  bool
}

type ChoicesPrompt struct {
	BasePrompt

	AddOMask  bool
	AddPrompt bool
	Length    int
}

type Config struct {
	ModelPath, ParamsPath string
	VocabFile             string
	DoLowerCase           bool
	MaxSeqLength          int
	ForCN                 bool
	// The maximum number of predictors for concurrent inferences.
	// Defaults to the value of runtime.NumCPU.
	MaxConcurrency int
}

type Encoding struct {
	SoftTokenIDs   []int
	InputIDs       []int
	PositionIDs    []int
	TokenTypeIDs   []int
	AttentionMask  *paddle.Matrix[float32]
	OMaskPositions []int
	ClsPositions   int
}

type ErnieTokenizer struct {
	*aietokenizer.Tokenizer
}

type Example struct {
	TextA    string
	TextB    string
	Question string
	Choices  []string
}

type Input struct {
	Text       string
	Positions  int
	TokenTypes int
	DoTruncate bool
}

type Prediction struct {
	Text   string
	Scores map[string]float32
}

type Prompt interface {
	BuildText(Example) string
	Positions() int
	TokenTypes() int
	DoTruncate() bool
}

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

type SepPrompt struct {
	BasePrompt

	Sep string
}

type TextPrompt struct {
	BasePrompt

	Text string
}

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