Documentation
¶
Index ¶
- Constants
- type Generator
- func (g *Generator) Cleanup() error
- func (g *Generator) Generate(imageName string) (weightsBase string, dockerfile string, dockerignoreContents string, err error)
- func (g *Generator) GenerateBase() (string, error)
- func (g *Generator) GenerateDockerfileWithoutSeparateWeights() (string, error)
- func (g *Generator) SetUseCudaBaseImage(argumentValue string)
Constants ¶
View Source
const DockerignoreHeader = `` /* 231-byte string literal not displayed */
Variables ¶
This section is empty.
Functions ¶
This section is empty.
Types ¶
type Generator ¶ added in v0.0.16
type Generator struct {
Config *config.Config
Dir string
// these are here to make this type testable
GOOS string
GOARCH string
// contains filtered or unexported fields
}
func (*Generator) Generate ¶ added in v0.0.16
func (g *Generator) Generate(imageName string) (weightsBase string, dockerfile string, dockerignoreContents string, err error)
Generate creates the Dockerfile and .dockerignore file contents for model weights It returns four values: - weightsBase: The base image used for Dockerfile generation for model weights. - dockerfile: A string that represents the Dockerfile content generated by the function. - dockerignoreContents: A string that represents the .dockerignore content. - err: An error object if an error occurred during Dockerfile generation; otherwise nil.
func (*Generator) GenerateBase ¶ added in v0.0.16
func (*Generator) GenerateDockerfileWithoutSeparateWeights ¶ added in v0.8.0
GenerateDockerfileWithoutSeparateWeights generates a Dockerfile that doesn't write model weights to a separate layer.
func (*Generator) SetUseCudaBaseImage ¶ added in v0.8.2
Source Files
Click to show internal directories.
Click to hide internal directories.