image

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, fastFlag bool, annotations map[string]string, localImage 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 ¶ added in v0.1.0

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 GeneratePipFreeze ¶ added in v0.13.3

func GeneratePipFreeze(imageName string, fastFlag bool) (string, error)

GeneratePipFreeze by running a pip freeze on the image. This will be run as part of the build process then added as a label to the image.

func GetConfig ¶ added in v0.0.14

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

func GetOpenAPISchema ¶ added in v0.1.0

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