From 80b933cd0383ee93a2a78d28c94cdfaab40f980f Mon Sep 17 00:00:00 2001
From: Biswaroop Bhattacharjee <biswaroop08@gmail.com>
Date: Mon, 3 Jul 2023 20:39:14 +0530
Subject: [PATCH] add: script for build-push of cht-xgen

---
 scripts/cht_xgen.sh | 14 ++++++++++++++
 1 file changed, 14 insertions(+)
 create mode 100755 scripts/cht_xgen.sh

diff --git a/scripts/cht_xgen.sh b/scripts/cht_xgen.sh
new file mode 100755
index 0000000..b280b4c
--- /dev/null
+++ 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