From 8b6f385e58ee83c1a43e3a89ee9b0562a00a5852 Mon Sep 17 00:00:00 2001
From: Richard Abrich <richard.abrich@gmail.com>
Date: Tue, 5 Mar 2024 15:08:32 -0500
Subject: [PATCH] add workflow file

---
 .github/workflows/docker-build-ec2.yml | 46 ++++++++++++++++++++++++++
 1 file changed, 46 insertions(+)
 create mode 100644 .github/workflows/docker-build-ec2.yml

diff --git a/.github/workflows/docker-build-ec2.yml b/.github/workflows/docker-build-ec2.yml
new file mode 100644
index 00000000..d34ad13e
--- /dev/null
+++ b/.github/workflows/docker-build-ec2.yml
@@ -0,0 +1,46 @@
+# Autogenerated via deploy.py, do not edit!
+
+name: Docker Build on EC2 Instance
+
+on:
+  push:
+    branches:
+      - feat/docker
+
+jobs:
+  build:
+    runs-on: ubuntu-latest
+    steps:
+      - name: Checkout code
+        uses: actions/checkout@v2
+
+      - name: SSH and Execute Build on EC2
+        uses: appleboy/ssh-action@master
+        with:
+          command_timeout: "60m"
+          host: 44.222.213.196
+          username: ubuntu  # Usually 'ubuntu' or 'ec2-user'
+          
+          key: ${{ secrets.SSH_PRIVATE_KEY }}
+          
+          script: |
+            source activate pytorch
+            nvidia-smi
+
+            rm -rf CogVLM || true
+            git clone https://github.com/OpenAdaptAI/CogVLM
+            cd CogVLM
+            git checkout feat/docker
+            git pull
+
+            # Stop and remove existing container if it's running
+            sudo docker stop cogvlm-container || true
+            sudo docker rm cogvlm-container || true
+
+            # Build the image
+            sudo nvidia-docker build -t cogvlm . || exit 1
+
+            # Run the image
+            sudo docker run -d -p 6092:6092 --gpus all --name cogvlm-container \
+              -e OPENAI_API_KEY=${{ secrets.OPENAI_API_KEY }} \
+              cogvlm
\ No newline at end of file