diff --git a/pkg/cli/build.go b/pkg/cli/build.go index 7b17fa74de..524c50dd0c 100644 --- a/pkg/cli/build.go +++ b/pkg/cli/build.go @@ -57,6 +57,9 @@ func buildCommand(cmd *cobra.Command, args []string) error { if err != nil { return err } + if cfg.Build.Fast { + buildFast = cfg.Build.Fast + } imageName := cfg.Image if buildTag != "" { diff --git a/pkg/cli/predict.go b/pkg/cli/predict.go index c3c325c54b..37e18e6404 100644 --- a/pkg/cli/predict.go +++ b/pkg/cli/predict.go @@ -76,6 +76,9 @@ func cmdPredict(cmd *cobra.Command, args []string) error { if err != nil { return err } + if cfg.Build.Fast { + buildFast = cfg.Build.Fast + } if imageName, err = image.BuildBase(cfg, projectDir, buildUseCudaBaseImage, DetermineUseCogBaseImage(cmd), buildProgressOutput); err != nil { return err @@ -117,6 +120,9 @@ func cmdPredict(cmd *cobra.Command, args []string) error { if gpus == "" && conf.Build.GPU { gpus = "all" } + if conf.Build.Fast { + buildFast = conf.Build.Fast + } } console.Info("") diff --git a/pkg/cli/push.go b/pkg/cli/push.go index eb65ec88ec..b1f50f1e83 100644 --- a/pkg/cli/push.go +++ b/pkg/cli/push.go @@ -45,6 +45,9 @@ func push(cmd *cobra.Command, args []string) error { if err != nil { return err } + if cfg.Build.Fast { + buildFast = cfg.Build.Fast + } imageName := cfg.Image if len(args) > 0 { diff --git a/pkg/cli/train.go b/pkg/cli/train.go index 725792c5f7..213b83115a 100644 --- a/pkg/cli/train.go +++ b/pkg/cli/train.go @@ -63,6 +63,9 @@ func cmdTrain(cmd *cobra.Command, args []string) error { if err != nil { return err } + if cfg.Build.Fast { + buildFast = cfg.Build.Fast + } if imageName, err = image.BuildBase(cfg, projectDir, buildUseCudaBaseImage, DetermineUseCogBaseImage(cmd), buildProgressOutput); err != nil { return err @@ -98,6 +101,9 @@ func cmdTrain(cmd *cobra.Command, args []string) error { if gpus == "" && conf.Build.GPU { gpus = "all" } + if conf.Build.Fast { + buildFast = conf.Build.Fast + } } console.Info("") diff --git a/pkg/config/config.go b/pkg/config/config.go index 68753f22f5..deeb455366 100644 --- a/pkg/config/config.go +++ b/pkg/config/config.go @@ -55,6 +55,7 @@ type Build struct { PreInstall []string `json:"pre_install,omitempty" yaml:"pre_install"` // Deprecated, but included for backwards compatibility CUDA string `json:"cuda,omitempty" yaml:"cuda"` CuDNN string `json:"cudnn,omitempty" yaml:"cudnn"` + Fast bool `json:"fast,omitempty" yaml:"fast"` pythonRequirementsContent []string }