huggingface

Documentation

Index

• Constants
• type HuggingFace
  • func New(model string, temperature float32, verbose bool) *HuggingFace
  • func (h *HuggingFace) Completion(ctx context.Context, prompt string) (string, error)
  • func (h *HuggingFace) WithMaxLength(maxLength int) *HuggingFace
  • func (h *HuggingFace) WithMinLength(minLength int) *HuggingFace
  • func (h *HuggingFace) WithMode(mode HuggingFaceMode) *HuggingFace
  • func (h *HuggingFace) WithModel(model string) *HuggingFace
  • func (h *HuggingFace) WithTemperature(temperature float32) *HuggingFace
  • func (h *HuggingFace) WithToken(token string) *HuggingFace
  • func (h *HuggingFace) WithTopK(topK int) *HuggingFace
  • func (h *HuggingFace) WithTopP(topP float32) *HuggingFace
  • func (h *HuggingFace) WithVerbose(verbose bool) *HuggingFace
• type HuggingFaceMode

Constants

const APIBaseURL = "https://api-inference.huggingface.co/models/"

const (
	ErrHuggingFaceCompletion = "huggingface completion error"
)

Types

type HuggingFace

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

func New

func New(model string, temperature float32, verbose bool) *HuggingFace

func (*HuggingFace) Completion

func (h *HuggingFace) Completion(ctx context.Context, prompt string) (string, error)

func (*HuggingFace) WithMaxLength

func (h *HuggingFace) WithMaxLength(maxLength int) *HuggingFace

func (*HuggingFace) WithMinLength

func (h *HuggingFace) WithMinLength(minLength int) *HuggingFace

func (*HuggingFace) WithMode

func (h *HuggingFace) WithMode(mode HuggingFaceMode) *HuggingFace

func (*HuggingFace) WithModel

func (h *HuggingFace) WithModel(model string) *HuggingFace

func (*HuggingFace) WithTemperature

func (h *HuggingFace) WithTemperature(temperature float32) *HuggingFace

func (*HuggingFace) WithToken

func (h *HuggingFace) WithToken(token string) *HuggingFace

func (*HuggingFace) WithTopK

func (h *HuggingFace) WithTopK(topK int) *HuggingFace

func (*HuggingFace) WithTopP

func (h *HuggingFace) WithTopP(topP float32) *HuggingFace

func (*HuggingFace) WithVerbose

func (h *HuggingFace) WithVerbose(verbose bool) *HuggingFace

type HuggingFaceMode

type HuggingFaceMode int

const (
	HuggingFaceModeCoversational HuggingFaceMode = iota
	HuggingFaceModeTextGeneration
)