image

Documentation

Index

• func Build(cfg *config.Config, dir, imageName string, secrets []string, ...) error
• func BuildBase(cfg *config.Config, dir string, useCudaBaseImage string, useCogBaseImage *bool, ...) (string, error)
• func GenerateOpenAPISchema(imageName string, enableGPU bool) (map[string]any, error)
• func GetConfig(imageName string) (*config.Config, error)
• func GetOpenAPISchema(imageName string) (*openapi3.T, error)

Functions

func Build

func Build(cfg *config.Config, dir, imageName string, secrets []string, noCache, separateWeights bool, useCudaBaseImage string, progressOutput string, schemaFile string, dockerfileFile string, useCogBaseImage *bool, strip bool, precompile bool) 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, useCudaBaseImage string, useCogBaseImage *bool, progressOutput string) (string, error)

func GenerateOpenAPISchema

func GenerateOpenAPISchema(imageName string, enableGPU bool) (map[string]any, error)

GenerateOpenAPISchema by running the image and executing Cog This will be run as part of the build process then added as a label to the image. It can be retrieved more efficiently with the label by using GetOpenAPISchema

func GetConfig

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

func GetOpenAPISchema

func GetOpenAPISchema(imageName string) (*openapi3.T, error)