update-s

sozercan · May 20, 2024 · cb16389 · cb16389
1 parent 6b21fbc
commit cb16389
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pkg/aikit2llb/inference/exllama.go b/pkg/aikit2llb/inference/exllama.go
@@ -24,7 +24,7 @@ func installExllama(c *config.InferenceConfig, s llb.State, merge llb.State) llb
 	// clone localai exllama backend only
 	s = cloneLocalAI(s)
 
-	s = s.Run(utils.Bashf("source $HOME/.cargo/env && export BUILD_TYPE=cublas && cd /tmp/localai/backend/python/%[1]s && ./install.sh", backend)).Root()
+	s = s.Run(utils.Bashf("source $HOME/.cargo/env && export BUILD_TYPE=cublas && cd /tmp/localai/backend/python/%[1]s && make exllama2", backend)).Root()
 
 	// clone exllama to localai exllama backend path and install python dependencies
 	// s = s.Run(utils.Bashf("git clone --depth 1 %[1]s --branch %[2]s /tmp/%[3]s && mv /tmp/%[3]s/* /tmp/localai/backend/python/%[3]s && rm -rf /tmp/%[3]s && cd /tmp/localai/backend/python/%[3]s && rm -rf .git && source $HOME/.cargo/env && python3 -m grpc_tools.protoc -I../.. --python_out=. --grpc_python_out=. backend.proto && uv venv && source .venv/bin/activate && ls -al && uv pip install --no-build-isolation --requirement requirements-install.txt && EXLLAMA_NOCOMPILE= uv pip install --no-build-isolation --requirement requirements.txt", exllamaRepo, exllamaTag, backend)).Root()