Fix weights-manifest change check

The weights are changed if there is no cache (no previous value) or the value differs. Signed-off-by: Iskren Chernev <[email protected]>
replicate · Aug 1, 2023 · f84143e · f84143e
1 parent 02c4a3e
commit f84143e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/pkg/image/build.go b/pkg/image/build.go
@@ -50,8 +50,8 @@ func Build(cfg *config.Config, dir, imageName string, secrets []string, noCache,
 		if err != nil {
 			return fmt.Errorf("Failed to generate weights manifest: %w", err)
 		}
-		cachedManifest, err := weights.LoadManifest(weightsManifestPath)
-		changed := err != nil && weightsManifest.Equal(cachedManifest)
+		cachedManifest, _ := weights.LoadManifest(weightsManifestPath)
+		changed := cachedManifest == nil || !weightsManifest.Equal(cachedManifest)
 		if changed {
 			if err := buildWeightsImage(dir, weightsDockerfile, imageName+"-weights", secrets, noCache, progressOutput); err != nil {
 				return fmt.Errorf("Failed to build model weights Docker image: %w", err)