From 41e51efc87edf045a6838836f70188de84ab4408 Mon Sep 17 00:00:00 2001
From: PAB <pierreantoine.bannier@gmail.com>
Date: Fri, 11 Oct 2024 17:38:29 +0200
Subject: [PATCH] chore(maint): add CI workflow to convert and generate a token
 (#192)

---
 .github/workflows/build.yml |  49 -------------
 .github/workflows/run.yml   | 134 ++++++++++++++++++++++++++++++++++++
 2 files changed, 134 insertions(+), 49 deletions(-)
 create mode 100644 .github/workflows/run.yml

diff --git a/.github/workflows/build.yml b/.github/workflows/build.yml
index 7d3ad49..326e2fa 100644
--- a/.github/workflows/build.yml
+++ b/.github/workflows/build.yml
@@ -36,55 +36,6 @@ env:
   GGML_N_THREADS: 1
 
 jobs:
-  ubuntu-latest-cmake:
-    runs-on: ubuntu-latest
-
-    steps:
-      - name: Clone
-        id: checkout
-        uses: actions/checkout@v4
-        with:
-          submodules: recursive
-
-      - name: Dependencies
-        id: depends
-        run: |
-          sudo apt-get update
-          sudo apt-get install build-essential
-
-      - name: Build
-        id: cmake_build
-        run: |
-          mkdir build
-          cd build
-          cmake ..
-          cmake --build . --config Release
-
-  macOS-latest-cmake:
-    runs-on: macos-latest
-
-    steps:
-      - name: Clone
-        id: checkout
-        uses: actions/checkout@v4
-        with:
-          submodules: recursive
-
-      - name: Dependencies
-        id: depends
-        continue-on-error: true
-        run: |
-          brew update
-
-      - name: Build
-        id: cmake_build
-        run: |
-          sysctl -a
-          mkdir build
-          cd build
-          cmake ..
-          cmake --build . --config Release
-
   windows-msys2:
     runs-on: windows-latest
 
diff --git a/.github/workflows/run.yml b/.github/workflows/run.yml
new file mode 100644
index 0000000..3ad75d4
--- /dev/null
+++ b/.github/workflows/run.yml
@@ -0,0 +1,134 @@
+name: run
+
+on:
+  push:
+    branches:
+      - main
+    paths:
+      [
+        ".github/workflows/**",
+        "**/CMakeLists.txt",
+        "**/Makefile",
+        "**/*.h",
+        "**/*.hpp",
+        "**/*.c",
+        "**/*.cpp",
+        "**/*.cu",
+      ]
+  pull_request:
+    types: [opened, synchronize, reopened]
+    paths:
+      [
+        ".github/workflows/**",
+        "**/CMakeLists.txt",
+        "**/Makefile",
+        "**/*.h",
+        "**/*.hpp",
+        "**/*.c",
+        "**/*.cpp",
+        "**/*.cu",
+      ]
+
+env:
+  BRANCH_NAME: ${{ github.head_ref || github.ref_name }}
+  GGML_NLOOP: 3
+  GGML_NITER: 1
+  GGML_N_THREADS: 4
+
+jobs:
+  ubuntu-latest-run:
+    runs-on: ubuntu-latest
+
+    steps:
+      - name: Clone
+        id: checkout
+        uses: actions/checkout@v4
+        with:
+          submodules: recursive
+
+      - name: Dependencies
+        id: depends
+        run: |
+          sudo apt-get update
+          sudo apt-get install build-essential python3.10 python3.10-venv python3.10-dev
+          python3.10 -m venv venv
+          source venv/bin/activate
+          python3 -m pip install -r requirements.txt
+
+      - name: Download
+        id: download
+        run: |
+          source venv/bin/activate
+          python3 download_weights.py --out-dir ./models --models bark-small
+
+      - name: Convert
+        id: convert
+        run: |
+          source venv/bin/activate
+          python3 convert.py --dir-model ./models/bark-small --use-f16
+
+      - name: Build
+        id: cmake_build
+        run: |
+          mkdir build
+          cd build
+          cmake ..
+          cmake --build . --config Release
+
+      - name: Run
+        id: load_and_run
+        run: >-
+          ./build/examples/main/main
+          -m ./models/bark-small/ggml_weights.bin
+          -p "hello"
+          -t 4
+
+  macOS-latest-run:
+    runs-on: macos-latest
+
+    steps:
+      - name: Clone
+        id: checkout
+        uses: actions/checkout@v4
+        with:
+          submodules: recursive
+
+      - name: Dependencies
+        id: depends
+        continue-on-error: true
+        run: |
+          brew update
+          brew install python@3.10
+          python3.10 -m pip install --upgrade pip
+          python3.10 -m venv venv
+          source venv/bin/activate
+          python3.10 -m pip install -r requirements.txt
+
+      - name: Download
+        id: download
+        run: |
+          source venv/bin/activate
+          python3 download_weights.py --out-dir ./models --models bark-small
+
+      - name: Convert
+        id: convert
+        run: |
+          source venv/bin/activate
+          python3 convert.py --dir-model ./models/bark-small --use-f16
+
+      - name: Build
+        id: cmake_build
+        run: |
+          sysctl -a
+          mkdir build
+          cd build
+          cmake ..
+          cmake --build . --config Release
+
+      - name: Run
+        id: load_and_run
+        run: >-
+          ./build/examples/main/main
+          -m ./models/bark-small/ggml_weights.bin
+          -p "hello"
+          -t 4