From 19669d0ac34d40d04bbfaba9b64d727615ddb477 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Serta=C3=A7=20=C3=96zercan?= <852750+sozercan@users.noreply.github.com> Date: Mon, 4 Dec 2023 22:16:37 -0800 Subject: [PATCH] chore: update to localai v2.0.0 (#22) Signed-off-by: Sertac Ozercan --- pkg/aikit2llb/convert.go | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/pkg/aikit2llb/convert.go b/pkg/aikit2llb/convert.go index b90b1d03..1b3d8e1f 100644 --- a/pkg/aikit2llb/convert.go +++ b/pkg/aikit2llb/convert.go @@ -15,7 +15,7 @@ import ( const ( debianSlim = "docker.io/library/debian:12-slim" distrolessBase = "gcr.io/distroless/cc-debian12:latest" - localAIVersion = "v1.40.0" + localAIVersion = "v2.0.0" cudaVersion = "12-3" ) @@ -101,7 +101,7 @@ func addLocalAI(c *config.Config, s llb.State, merge llb.State) (llb.State, llb. var localAIURL string switch c.Runtime { case utils.RuntimeNVIDIA: - localAIURL = fmt.Sprintf("https://sertaccdn.azureedge.net/localai/%s/local-ai", localAIVersion) + localAIURL = fmt.Sprintf("https://sertaccdn.azureedge.net/localai/%s/cuda12/local-ai", localAIVersion) case utils.RuntimeCPUAVX2: localAIURL = fmt.Sprintf("https://github.com/mudler/LocalAI/releases/download/%s/local-ai-avx2-Linux-x86_64", localAIVersion) case utils.RuntimeCPUAVX512: