domain

Documentation

Index

• Variables
• func Init(cfg *Config)
• type Account
  • func NewAccount(v string) (Account, error)
• type Config
  • func (r *Config) SetDefault()
• type Finetune
• type FinetuneConfig
• type FinetuneName
  • func NewFinetuneName(v string) (FinetuneName, error)
• type FinetuneParameter
  • func NewFinetuneParameter(model, task string, hyperparameters map[string]string) (FinetuneParameter, error)
• type FinetuneParameterConfig
• type FinetuneStatus
• type JobDetail
• type JobInfo

Variables

var (
	FinetuneStatusFailed      = finetuneStatus("Failed")
	FinetuneStatusPending     = finetuneStatus("Pending")
	FinetuneStatusRunning     = finetuneStatus("Running")
	FinetuneStatusCreating    = finetuneStatus("Creating")
	FinetuneStatusAbnormal    = finetuneStatus("Abnormal")
	FinetuneStatusCompleted   = finetuneStatus("Completed")
	FinetuneStatusTerminated  = finetuneStatus("Terminated")
	FinetuneStatusTerminating = finetuneStatus("Terminating")
)

Functions

func Init

func Init(cfg *Config)

Types

type Account

type Account interface {
	Account() string
}

Account

func NewAccount

func NewAccount(v string) (Account, error)

type Config

type Config struct {
	MaxFinetuneNameLength int `json:"max_finetune_name_length"`
	MinFinetuneNameLength int `json:"min_finetune_name_length"`

	// Key is the finetue model name
	Finetunes map[string]FinetuneParameterConfig `json:"finetunes" required:"true"`
}

func (*Config) SetDefault

func (r *Config) SetDefault()

type Finetune

type Finetune struct {
	Id string

	FinetuneConfig
}

type FinetuneConfig

type FinetuneConfig struct {
	Name  FinetuneName
	Param FinetuneParameter
}

type FinetuneName

type FinetuneName interface {
	FinetuneName() string
}

FinetuneName

func NewFinetuneName

func NewFinetuneName(v string) (FinetuneName, error)

type FinetuneParameter

type FinetuneParameter interface {
	Model() string
	Task() string
	Hypeparameters() map[string]string
}

func NewFinetuneParameter

func NewFinetuneParameter(model, task string, hyperparameters map[string]string) (
	FinetuneParameter, error,
)

type FinetuneParameterConfig

type FinetuneParameterConfig struct {
	Tasks           []string `json:"tasks"           required:"true"`
	Hyperparameters []string `json:"hyperparameters" required:"true"`
}

type FinetuneStatus

type FinetuneStatus interface {
	FinetuneStatus() string
	IsDone() bool
	IsSuccess() bool
}

FinetuneStatus

type JobDetail

type JobDetail struct {
	Status   FinetuneStatus
	Duration int
}

type JobInfo

type JobInfo struct {
	JobId string
}