updated model ids

premAI-io · Sep 20, 2023 · beb0765 · beb0765
1 parent 2599e7f
commit beb0765
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/cht-petals/build.sh b/cht-petals/build.sh
@@ -3,5 +3,5 @@ set -e
 export VERSION=1.0.4
 source "$(dirname "${BASH_SOURCE[0]}")/../utils.sh"
 
-build_cpu ghcr.io/premai-io/chat-codellama-34b-cpu   codellama-34b   ${@:1}
-build_cpu ghcr.io/premai-io/chat-stable-beluga-2-cpu stable-beluga-2 ${@:1}
+build_cpu ghcr.io/premai-io/chat-codellama-34b-cpu   premai-io/CodeLlama-34b-Instruct-hf ${@:1}
+build_cpu ghcr.io/premai-io/chat-stable-beluga-2-cpu petals-team/StableBeluga2           ${@:1}