data

Documentation

Index

• Variables
• func AddDiscordInputTask(name string, reference *discordgo.MessageReference, task *dfpb.Task)
• func AddImageMetaData(r io.Reader, imgdesc string, producer string) (io.Reader, error)
• func CreateGptNeoTask(prompt string) *dfpb.Task
• func CreateImageUpscaleInputTask(data *[]byte) *dfpb.Input
• func CreateInterrogatorInputTask(data *[]byte) *dfpb.Input
• func CreateSDTaskWithCmdArgs(args *CommandArgs, imgbuff *bytes.Buffer, initimgurl string, removeseed bool) (*dfpb.Task, string)
• func CreateTask(il []*dfpb.Input, ol []*dfpb.Output) *dfpb.Task
• func DownloadFile(url string) (*bytes.Buffer, error)
• func JsonToTask(body *[]byte) (*dfpb.Task, error)
• func TaskNameToQNameAndRKey(taskname string) (string, string)
• type AISettings
• type CommandArgs
  • func ArgsParse(input string, argslist []string) *CommandArgs
• type DiffSettings
• type GptNeoSettings
• type InputTask
• type Pstatus
• type QueueItem
• type RealEsrganSettings
• type UserInfo
• type WorkerItem

Variables

var ArgsList []string = []string{
	"-h", "--help", "--tokenize", "-t", "--height", "-H", "--width", "-W",
	"--cfg_scale", "-C", "--number", "-n", "--separate-images", "-i", "--grid", "-g",
	"--strength", "-E", "--init_image_url", "-IMG",
	"--sampler", "-A", "--steps", "-s", "--seed", "-S", "--prior", "-p", "--upscale", "-U", "--face",
	"--model", "-M"}

Functions

func AddDiscordInputTask

func AddDiscordInputTask(name string, reference *discordgo.MessageReference, task *dfpb.Task)

func AddImageMetaData

func AddImageMetaData(r io.Reader, imgdesc string, producer string) (io.Reader, error)

func CreateGptNeoTask

func CreateGptNeoTask(prompt string) *dfpb.Task

func CreateImageUpscaleInputTask

func CreateImageUpscaleInputTask(data *[]byte) *dfpb.Input

func CreateInterrogatorInputTask

func CreateInterrogatorInputTask(data *[]byte) *dfpb.Input

func CreateSDTaskWithCmdArgs

func CreateSDTaskWithCmdArgs(args *CommandArgs, imgbuff *bytes.Buffer, initimgurl string, removeseed bool) (*dfpb.Task, string)

func CreateTask

func CreateTask(il []*dfpb.Input, ol []*dfpb.Output) *dfpb.Task

func DownloadFile

func DownloadFile(url string) (*bytes.Buffer, error)

func JsonToTask

func JsonToTask(body *[]byte) (*dfpb.Task, error)

func TaskNameToQNameAndRKey

func TaskNameToQNameAndRKey(taskname string) (string, string)

Types

type AISettings

type AISettings struct {
	AiName   string      `json:"ainame"`
	Settings interface{} `json:"settings"`
}

type CommandArgs

type CommandArgs struct {
	Cmd  string
	Args []string
}

func ArgsParse

func ArgsParse(input string, argslist []string) *CommandArgs

type DiffSettings

type DiffSettings struct {
	Pipeline          string  `json:"pipeline"`
	Prompt            string  `json:"prompt"`
	Seed              uint64  `json:"seed"`
	Number            uint    `json:"number"`
	Height            uint    `json:"height"`
	Width             uint    `json:"width"`
	Strength          float32 `json:"strength"`
	NumInferenceSteps uint    `json:"num_inference_steps"`
	Guidance_scale    float32 `json:"guidance_scale"`
	InitImageUrl      string  `json:"init_image_url"`
	Eta               float32 `json:"eta"`
}

type GptNeoSettings

type GptNeoSettings struct {
	Prompt string `json:"prompt"`
}

type InputTask

type InputTask struct {
	Id         string      `json:"id"`
	User       *UserInfo   `json:"user"`
	AiSettings *AISettings `json:"aisettings"`
}

type Pstatus

type Pstatus int

const (
	StatusCmd Pstatus = iota
	StatusArgPrefix
	StatusArgSkip
	StatusArgItemKStart
	StatusArgItemVStart
	StatusArgItemEnd
)

type QueueItem

type QueueItem struct {
	Name     string
	Bindkeys []string
}

type RealEsrganSettings

type RealEsrganSettings struct {
	ModelType   string `json:"model_type"`
	FaceEnhance bool   `json:"face_enhance"`
}

type UserInfo

type UserInfo struct {
	Provider       string `json:"provider"`
	ProviderUserId string `json:"provideruserid"`
	ProviderData   string `json:providerdata"`
}

type WorkerItem

type WorkerItem struct {
	Name     string
	Bindkeys []string
}