From f9881fb780292304abfd46fc2a65fa9200dc78c8 Mon Sep 17 00:00:00 2001 From: Biswaroop Bhattacharjee Date: Thu, 29 Jun 2023 01:12:01 +0530 Subject: [PATCH] update: script for falcon-7b-instruct image --- scripts/cht_falcon.sh | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/scripts/cht_falcon.sh b/scripts/cht_falcon.sh index f8d4159..8dcf2d1 100644 --- 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