image

Documentation

Index

• func Build(cfg *config.Config, dir, imageName string, progressOutput string) error
• func BuildBase(cfg *config.Config, dir string, progressOutput string) (string, error)
• func GetConfig(imageName string) (*config.Config, error)
• type Input
• type InputType
• type TypeSignature
  • func GetTypeSignature(imageName string, enableGPU bool) (*TypeSignature, error)

Functions

func Build

func Build(cfg *config.Config, dir, imageName string, progressOutput string) error

Build a Cog model from a config

This is separated out from docker.Build(), so that can be as close as possible to the behavior of 'docker build'.

func BuildBase

func BuildBase(cfg *config.Config, dir string, progressOutput string) (string, error)

func GetConfig

func GetConfig(imageName string) (*config.Config, error)

Types

type Input

type Input struct {
	Name    string    `json:"name"`
	Type    InputType `json:"type,omitempty"`
	Default *string   `json:"default,omitempty"`
	Min     *string   `json:"min,omitempty"`
	Max     *string   `json:"max,omitempty"`
	Options *[]string `json:"options,omitempty"`
	Help    *string   `json:"help,omitempty"`
}

type InputType

type InputType string

const (
	InputTypeString InputType = "str"
	InputTypeInt    InputType = "int"
	InputTypeFloat  InputType = "float"
	InputTypeBool   InputType = "bool"
	InputTypePath   InputType = "Path"
)

type TypeSignature

type TypeSignature struct {
	Inputs []Input `json:"inputs,omitempty"`
}

func GetTypeSignature

func GetTypeSignature(imageName string, enableGPU bool) (*TypeSignature, error)