feat: special case runner for large models (#16)

Author: bo0tzz

.github/scripts/scope.js

@@ -28,7 +28,8 @@ module.exports = ({core}) => {
     if (!m || !n) return false;
     return m["name"] === n["name"] &&
       m["source"] === n["source"] &&
-      m["hf-name"] === n["hf-name"];
+      m["hf-name"] === n["hf-name"] &&
+      m["runner"] === n["runner"];
   }
 
   for (const key of keys) {

.github/workflows/export.yaml

@@ -10,6 +10,9 @@ on:
       model-source:
         required: true
         type: string
+      runner:
+        required: true
+        type: string
       hf-name:
         required: false
         type: string
@@ -19,13 +22,13 @@ on:
 
 jobs:
   export:
-    runs-on: ubuntu-latest
+    runs-on: ${{ inputs.runner }}
     steps:
       - name: Checkout
         uses: actions/checkout@v4
 
       - uses: astral-sh/setup-uv@v5
-      - run: uv sync # TODO: cache uv env (does setup-uv do that already automatically?)
+      - run: uv sync
 
       - run: uv run immich_model_exporter export "${{ inputs.model-name }}" "${{ inputs.model-source }}" --hf-model-name "${{ inputs.hf-name }}"
       - uses: actions/upload-artifact@v4
@@ -38,14 +41,14 @@ jobs:
 
   upload:
     if: ${{ inputs.upload }}
-    runs-on: ubuntu-latest
+    runs-on: ${{ inputs.runner }}
     needs: export
     steps:
       - name: Checkout
         uses: actions/checkout@v4
 
       - uses: astral-sh/setup-uv@v5
-      - run: uv sync # TODO: cache uv env (does setup-uv do that already automatically?)
+      - run: uv sync
 
       - uses: actions/download-artifact@v4
         with:

.github/workflows/push.yaml

@@ -81,4 +81,5 @@ jobs:
       model-name: ${{ matrix.name }}
       model-source: ${{ matrix.source }}
       hf-name: ${{ matrix.hf-name }}
+      runner: ${{ matrix.runner || 'ubuntu-latest'}}
       upload: ${{ inputs.force || github.event_name == 'release'}}

models.yaml

@@ -99,6 +99,7 @@ models:
     source: 'openclip'
   - name: 'ViT-SO400M-14-SigLIP-384__webli'
     source: 'openclip'
+    runner: 'mich'
   - name: 'ViT-SO400M-14-SigLIP2-378__webli'
     source: 'openclip'
   - name: 'ViT-SO400M-14-SigLIP2__webli'
@@ -107,23 +108,30 @@ models:
     source: 'openclip'
   - name: 'ViT-SO400M-16-SigLIP2-384__webli'
     source: 'openclip'
+    runner: 'mich'
   - name: 'ViT-SO400M-16-SigLIP2-512__webli'
     source: 'openclip'
+    runner: 'mich'
   - name: 'ViT-gopt-16-SigLIP2-256__webli'
     source: 'openclip'
+    runner: 'mich'
   - name: 'ViT-gopt-16-SigLIP2-384__webli'
     source: 'openclip'
+    runner: 'mich'
   - name: 'nllb-clip-base-siglip__mrl'
     source: 'openclip'
+    runner: 'mich'
   - name: 'nllb-clip-base-siglip__v1'
     source: 'openclip'
   - name: 'nllb-clip-large-siglip__mrl'
     source: 'openclip'
   - name: 'nllb-clip-large-siglip__v1'
     source: 'openclip'
+    runner: 'mich'
   - name: 'XLM-Roberta-Base-ViT-B-32__laion5b_s13b_b90k'
     hf-name: 'xlm-roberta-base-ViT-B-32__laion5b_s13b_b90k'
     source: 'openclip'
-  - name: 'XML-Roberta-Large-ViT-H-14__frozen_laion5b_s13b_b90k'
+  - name: 'XLM-Roberta-Large-ViT-H-14__frozen_laion5b_s13b_b90k'
     hf-name: 'xlm-roberta-large-ViT-H-14__frozen_laion5b_s13b_b90k'
     source: 'openclip'
+    runner: 'mich'