update: script for falcon-7b-instruct image

premAI-io · Jun 28, 2023 · f9881fb · f9881fb
1 parent 7de5f92
commit f9881fb
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/scripts/cht_falcon.sh b/scripts/cht_falcon.sh
@@ -5,10 +5,10 @@ set -e
 export VERSION=1.0.0
 
 docker buildx build --push \
-    --cache-from ghcr.io/premai-io/chat-falcon-7b-gpu:latest \
+    --cache-from ghcr.io/premai-io/chat-falcon-7b-instruct-gpu:latest \
     --file ./cht-falcon/docker/gpu/Dockerfile \
-    --build-arg="MODEL_ID=tiiuae/falcon-7b" \
-    --tag ghcr.io/premai-io/chat-falcon-7b-gpu:latest \
-    --tag ghcr.io/premai-io/chat-falcon-7b-gpu:$VERSION \
+    --build-arg="MODEL_ID=tiiuae/falcon-7b-instruct" \
+    --tag ghcr.io/premai-io/chat-falcon-7b-instruct-gpu:latest \
+    --tag ghcr.io/premai-io/chat-falcon-7b-instruct-gpu:$VERSION \
     ./cht-falcon
-docker run --gpus all ghcr.io/premai-io/chat-falcon-7b-gpu:latest pytest
+docker run --gpus all ghcr.io/premai-io/chat-falcon-7b-instruct-gpu:latest pytest