mutators

Documentation

Index

• Constants
• Variables
• func ErrWrongNumberOfArgs(min, max, got int) error
• func J2Y(w io.WriteCloser, r io.ReadCloser, _ any) (int64, error)
• func Y2J(w io.WriteCloser, r io.ReadCloser, _ any) (int64, error)
• type HuggingFaceRequest
• type Response

Constants

const (
	WrapMaxChars = 80
	IndentChars  = 4
)

Variables

var HuggingFaceCommonTasks = map[string]string{

	"fillmask":        "bert-base-uncased",
	"summarization":   "facebook/bart-large-cnn",
	"classification":  "distilbert-base-uncased-finetuned-sst-2-english",
	"text-generation": "gpt2",
	"ner":             "dbmdz/bert-large-cased-finetuned-conll03-english",
	"translation":     "t5-base",
	"bloom":           "bigscience/bloom",
	"bloomz":          "bigscience/bloomz",
	"chat":            "OpenAssistant/oasst-sft-4-pythia-12b-epoch-3.5",
	"starcoder":       "bigcode/starcoder",
}

Functions

func ErrWrongNumberOfArgs

func ErrWrongNumberOfArgs(min, max, got int) error

func J2Y

func J2Y(w io.WriteCloser, r io.ReadCloser, _ any) (int64, error)

func Y2J

func Y2J(w io.WriteCloser, r io.ReadCloser, _ any) (int64, error)

Types

type HuggingFaceRequest

type HuggingFaceRequest struct {
	Inputs  string         `json:"inputs"`
	Options map[string]any `json:"options"`
}

type Response

type Response struct {
	Data struct {
		Translations []struct {
			TranslatedText         string
			DetectedSourceLanguage string
		}
	}
}