feat: Auto image build (#67)

Signed-off-by: Ishaan Sehgal <ishaanforthewin@gmail.com>
kaito-project · Oct 18, 2023 · 4df4266 · 4df4266
1 parent ce26b36
commit 4df4266
Show file tree

Hide file tree

Showing 10 changed files with 471 additions and 189 deletions.
diff --git a/.github/matrix-configs.json b/.github/matrix-configs.json
@@ -0,0 +1,100 @@
+[
+    {
+        "model": {
+            "runs_on": "self-hosted",
+            "name": "falcon-7b",
+            "dockerfile": "docker/presets/falcon/Dockerfile",
+            "build_args": "--build-arg FALCON_MODEL_NAME=tiiuae/falcon-7b"
+        },
+        "shouldBuildFalcon": "true"
+    },
+    {
+        "model": {
+            "runs_on": "self-hosted",
+            "name": "falcon-7b-instruct",
+            "dockerfile": "docker/presets/falcon/Dockerfile",
+            "build_args": "--build-arg FALCON_MODEL_NAME=tiiuae/falcon-7b-instruct"
+        },
+        "shouldBuildFalcon": "true"
+    },
+
+    {
+        "model": {
+            "runs_on": "self-hosted",
+            "name": "falcon-40b",
+            "dockerfile": "docker/presets/falcon/Dockerfile",
+            "build_args": "--build-arg FALCON_MODEL_NAME=tiiuae/falcon-40b"
+        },
+        "shouldBuildFalcon": "true"
+    },
+
+    {
+        "model": {
+            "runs_on": "self-hosted",
+            "name": "falcon-40b-instruct",
+            "dockerfile": "docker/presets/falcon/Dockerfile",
+            "build_args": "--build-arg FALCON_MODEL_NAME=tiiuae/falcon-40b-instruct"
+        },
+        "shouldBuildFalcon": "true"
+    },
+
+    {
+        "model": {
+            "runs_on": "self-hosted",
+            "name": "llama-2-7b",
+            "dockerfile": "docker/presets/llama-2/Dockerfile",
+            "build_args": "--build-arg LLAMA_VERSION=llama-2-7b --build-arg SRC_DIR=/home/presets/llama-2"
+        },
+        "shouldBuildLlama2": "true"
+    },
+
+    {
+        "model": {
+            "runs_on": "self-hosted",
+            "name": "llama-2-13b",
+            "dockerfile": "docker/presets/llama-2/Dockerfile",
+            "build_args": "--build-arg LLAMA_VERSION=llama-2-13b --build-arg SRC_DIR=/home/presets/llama-2"
+        },
+        "shouldBuildLlama2": "true"
+    },
+
+    {
+        "model": {
+            "runs_on": "self-hosted",
+            "name": "llama-2-70b",
+            "dockerfile": "docker/presets/llama-2/Dockerfile",
+            "build_args": "--build-arg LLAMA_VERSION=llama-2-70b --build-arg SRC_DIR=/home/presets/llama-2"
+        },
+        "shouldBuildLlama2": "true"
+    },
+
+    {
+        "model": {
+            "runs_on": "self-hosted",
+            "name": "llama-2-7b-chat",
+            "dockerfile": "docker/presets/llama-2/Dockerfile",
+            "build_args": "--build-arg LLAMA_VERSION=llama-2-7b-chat --build-arg SRC_DIR=/home/presets/llama-2-chat"
+        },
+        "shouldBuildLlama2Chat": "true"
+    },
+
+    {
+        "model": {
+            "runs_on": "self-hosted",
+            "name": "llama-2-13b-chat",
+            "dockerfile": "docker/presets/llama-2/Dockerfile",
+            "build_args": "--build-arg LLAMA_VERSION=llama-2-13b-chat --build-arg SRC_DIR=/home/presets/llama-2-chat"
+        },
+        "shouldBuildLlama2Chat": "true"
+    },
+
+    {
+        "model": {
+            "runs_on": "self-hosted",
+            "name": "llama-2-70b-chat",
+            "dockerfile": "docker/presets/llama-2/Dockerfile",
+            "build_args": "--build-arg LLAMA_VERSION=llama-2-70b-chat --build-arg SRC_DIR=/home/presets/llama-2-chat"
+        },
+        "shouldBuildLlama2Chat": "true"
+    }
+]