From 30bee42e16fb94f8ccbee5d97ca9712daa32e7c2 Mon Sep 17 00:00:00 2001 From: greg pereira Date: Mon, 3 Jun 2024 08:21:42 -0700 Subject: [PATCH] lamacpp-python-vulkan for arm64 Signed-off-by: greg pereira --- .github/workflows/model_servers.yaml | 11 ++++++++--- 1 file changed, 8 insertions(+), 3 deletions(-) diff --git a/.github/workflows/model_servers.yaml b/.github/workflows/model_servers.yaml index 9178a45a..7a9cfa6e 100644 --- a/.github/workflows/model_servers.yaml +++ b/.github/workflows/model_servers.yaml @@ -32,12 +32,17 @@ jobs: directory: llamacpp_python platforms: linux/amd64,linux/arm64 no_gpu: 1 - - image_name: llamacpp_python_cuda + - image_name: llamacpp-python-cuda model: granite flavor: cuda directory: llamacpp_python - platforms: linux/amd64 - cuda: 1 + platforms: linux/amd64,linux/arm64 + - image_name: llamacpp-python-vulkan + model: granite + flavor: vulkan + directory: llamacpp_python + platforms: linux/arm64 + no_gpu: 1 - image_name: whispercpp model: whisper-small flavor: base