image

Documentation

Index

• func Build(ctx context.Context, cfg *config.Config, dir, imageName string, ...) (string, error)
• func BuildAddLabelsAndSchemaToImage(ctx context.Context, dockerClient command.Command, tmpName, image string, ...) (string, error)
• func BuildBase(ctx context.Context, dockerClient command.Command, cfg *config.Config, ...) (string, error)
• func CogConfigFromManifest(ctx context.Context, manifest *image.InspectResponse) (*config.Config, error)
• func GenerateOpenAPISchema(ctx context.Context, dockerClient command.Command, imageName string, ...) (map[string]any, error)
• func GeneratePipFreeze(ctx context.Context, dockerClient command.Command, imageName string) (string, error)

Functions

func Build

func Build(
	ctx context.Context,
	cfg *config.Config,
	dir,
	imageName string,
	configFilename string,
	secrets []string,
	noCache,
	separateWeights bool,
	useCudaBaseImage string,
	progressOutput string,
	schemaFile string,
	dockerfileFile string,
	useCogBaseImage *bool,
	strip bool,
	precompile bool,
	annotations map[string]string,
	dockerCommand command.Command,
	client registry.Client) (string, error)

Build a Cog model from a config and returns the image ID (sha256:...) on success.

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

func BuildAddLabelsAndSchemaToImage

func BuildAddLabelsAndSchemaToImage(ctx context.Context, dockerClient command.Command, tmpName, image string, labels map[string]string, bundledSchemaFile string, progressOutput string) (string, error)

BuildAddLabelsAndSchemaToImage builds a cog model with labels and schema. Returns the image ID (sha256:...) of the final image.

The new image is based on the provided image with the labels and schema file appended to it. tmpName is the source image to build from, image is the final image name/tag.

func BuildBase

func BuildBase(ctx context.Context, dockerClient command.Command, cfg *config.Config, dir string, configFilename string, useCudaBaseImage string, useCogBaseImage *bool, progressOutput string, client registry.Client, requiresCog bool) (string, error)

func CogConfigFromManifest

func CogConfigFromManifest(ctx context.Context, manifest *image.InspectResponse) (*config.Config, error)

func GenerateOpenAPISchema

func GenerateOpenAPISchema(ctx context.Context, dockerClient command.Command, 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

func GeneratePipFreeze(ctx context.Context, dockerClient command.Command, imageName string) (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.