Fix loading of default params

Author: victorges

.github/workflows/ai-runner-docker-live-streamdiffusion.yaml

@@ -74,22 +74,6 @@ jobs:
           username: ${{ secrets.CI_DOCKERHUB_USERNAME }}
           password: ${{ secrets.CI_DOCKERHUB_TOKEN }}
 
-      - name: Load params for subtype
-        id: params_json
-        shell: bash
-        run: |
-          set -ex
-
-          PARAM_NAME="$(echo "${{ matrix.subtype }}" | tr '-' '_')"
-          JSON_FILE="live/streamdiffusion/default_params/${PARAM_NAME}.json"
-          if [ ! -f "$JSON_FILE" ]; then
-            echo "Params file missing: $JSON_FILE" >&2
-            exit 1
-          fi
-
-          JSON_COMPACT=$(tr -d '\n' < "$JSON_FILE")
-          echo "json_compact=$JSON_COMPACT" >> "$GITHUB_OUTPUT"
-
       - name: Build and push pipeline app image
         id: build-app
         uses: docker/build-push-action@v6
@@ -102,7 +86,7 @@ jobs:
           build-args: |
             GIT_SHA=${{ (github.ref_type == 'tag' && github.ref_name) || (github.event.pull_request.head.sha || github.sha) }}
             VERSION=${{ steps.version.outputs.version }}
-            PIPE_INITIAL_PARAMS=${{ steps.params_json.outputs.json_compact }}
+            SUBVARIANT=${{ matrix.subtype }}
           platforms: linux/amd64
           tags: ${{ steps.meta.outputs.tags }}
           annotations: ${{ steps.meta.outputs.annotations }}

live/streamdiffusion/Dockerfile

@@ -20,6 +20,7 @@ RUN uv run python -m streamdiffusion.tools.install-tensorrt
 COPY runner/app/ /app/runner/app/
 COPY runner/images/ /app/runner/images/
 COPY live/streamdiffusion/pipeline ./pipeline/
+COPY live/streamdiffusion/default_params ./default_params/
 COPY live/streamdiffusion/main.py ./main.py
 
 # Need --inexact to keep packages installed from install-tensorrt
@@ -38,10 +39,10 @@ ENV HF_HUB_ENABLE_HF_TRANSFER=1 \
 
 ARG GIT_SHA
 ARG VERSION="undefined"
-ARG PIPE_INITIAL_PARAMS
+ARG SUBVARIANT="sdturbo"
 
 ENV GIT_SHA="${GIT_SHA}" \
     VERSION="${VERSION}" \
-    PIPE_INITIAL_PARAMS="${PIPE_INITIAL_PARAMS}"
+    SUBVARIANT="${SUBVARIANT}"
 
 CMD ["uv", "run", "python", "main.py"]

live/streamdiffusion/main.py

@@ -1,14 +1,26 @@
 import json
 import os
+from pathlib import Path
 
 from app.app import start_app
 from app.pipelines.live_video_to_video import LiveVideoToVideoPipeline
 from app.live.pipelines import PipelineSpec
 
-initial_params = json.loads(os.environ.get("PIPE_INITIAL_PARAMS", "{}"))
+initial_params = {}
+
+subvariant = os.environ.get("SUBVARIANT")
+if subvariant:
+    params_filename = subvariant.replace("-", "_") + ".json"
+    params_path = Path(__file__).parent / "default_params" / params_filename
+
+    with open(params_path) as f:
+        initial_params = json.load(f)
+
+# Use subvariant as suffix unless it's sdturbo (default)
+name_suffix = "" if subvariant == "sdturbo" else f"-{subvariant}"
 
 pipeline_spec = PipelineSpec(
-    name="streamdiffusion",
+    name=f"streamdiffusion{name_suffix}",
     pipeline_cls="pipeline.pipeline:StreamDiffusion",
     params_cls="pipeline.params:StreamDiffusionParams",
     initial_params=initial_params,

runner/app/live/pipelines/spec.py

@@ -25,7 +25,10 @@ class PipelineSpec(BaseModel):
         ... )
     """
     name: str = ""
-    """Identifier for the pipeline. Derived from pipeline_cls if not provided."""
+    """
+    Identifier for the pipeline. Derived from pipeline_cls if not provided.
+    This must match the MODEL_ID set by the Orchestrator for the runner container.
+    """
 
     pipeline_cls: str
     """Import path to the Pipeline subclass. e.g. "my_pipelines.module:MyPipeline" """