cht-petals: build only arm64, skip tests

TODO: revert when Petals fixes this
premAI-io · Sep 20, 2023 · 01c62ac · 01c62ac
1 parent d1ba345
commit 01c62ac
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/cht-petals/build.sh b/cht-petals/build.sh
@@ -1,7 +1,10 @@
 #!/bin/bash
 set -e
-export VERSION=1.0.4
+export VERSION=1.0.0
 source "$(dirname "${BASH_SOURCE[0]}")/../utils.sh"
 
-build_cpu ghcr.io/premai-io/chat-codellama-34b-cpu   premai-io/CodeLlama-34b-Instruct-hf ${@:1}
+# TODO: support linux/amd64
+BUILDX_PLATFORM=linux/arm64 TESTS_SKIP_CPU=1 \
 build_cpu ghcr.io/premai-io/chat-stable-beluga-2-cpu petals-team/StableBeluga2           ${@:1}
+BUILDX_PLATFORM=linux/arm64 TESTS_SKIP_CPU=1 \
+build_cpu ghcr.io/premai-io/chat-codellama-34b-cpu   premai-io/CodeLlama-34b-Instruct-hf ${@:1}