fix: set config file env (#401)

Signed-off-by: Sertac Ozercan <[email protected]>
sozercan · Sep 27, 2024 · b4d52b5 · b4d52b5
1 parent a1c0cb1
commit b4d52b5
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/pkg/aikit2llb/inference/image.go b/pkg/aikit2llb/inference/image.go
@@ -34,6 +34,7 @@ func emptyImage(c *config.InferenceConfig, platform *specs.Platform) *specs.Imag
 
 	img.Config.Env = []string{
 		"PATH=" + system.DefaultPathEnv(utils.PlatformLinux),
+		"CONFIG_FILE=/config.yaml",
 	}
 
 	cudaEnv := []string{