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) 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)