add: script for build-push of cht-xgen

premAI-io · Jul 3, 2023 · 80b933c · 80b933c
1 parent a3d9f62
commit 80b933c
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/scripts/cht_xgen.sh b/scripts/cht_xgen.sh
@@ -0,0 +1,14 @@
+#!/bin/bash
+
+set -e
+
+export VERSION=1.0.0
+
+docker buildx build \
+    --cache-from ghcr.io/premai-io/chat-xgen-7b-8k-inst-gpu:latest \
+    --file ./cht-xgen/docker/gpu/Dockerfile \
+    --build-arg="MODEL_ID=Salesforce/xgen-7b-8k-inst" \
+    --tag ghcr.io/premai-io/chat-xgen-7b-8k-inst-gpu:latest \
+    --tag ghcr.io/premai-io/chat-xgen-7b-8k-inst-gpu:$VERSION \
+    ./cht-xgen
+docker run --gpus all ghcr.io/premai-io/chat-xgen-7b-8k-inst-gpu:latest pytest